how to work with matrix in numpy