create a matrix in python