How To Code A 2d Array Python