how to print matrix in python