how to generate a matrix in python