How to create a 2-dimensional array in Python