Python 将3D NumPy数组转换为2D，并从3D数组中获取两个不同的2D数组

我有一个3D NumPy阵列（2700、4000、2），我想将其转换为两个不同的2D阵列。我的意思是，我的新两个阵列将是（27004000）

我是Python编程新手。有办法吗？

IIUC，给你

代码

#Define the 3D array
n1 = np.ones((2700,4000,2))
#Get the first 2D array
n2 = n1[:,:,0]
#Get the second 2D array
n3 = n1[:,:,1]
print("Original Shape : ", n1.shape)
print("First array shape : ", n2.shape)
print("Second array shape : ", n3.shape) 

输出

Original Shape :  (2700, 4000, 2)
First array shape :  (2700, 4000)
Second array shape :  (2700, 4000)

---

假设你想说2

（27004000）

数组。如果您解释一下为什么您更喜欢两个2D阵列，可能会有所帮助。