Writing Matrix In Python