How To Create A 3d Array In Python