make matrix in Python