Creating A 3d Array In Python