How to define a two-dimensional array in Python