how to work with matrix in python numpy