[python] How to make matrices in Python?