Creating 3d Array In Python