Python 如何从numpy数组中提取所需维度

假设我有一个三维[3500500]的numpy数组，如何将一维提取为[1500500]

import numpy as np
my_array = np.ones((3,500,500),dtype=int,order='C')
print (my_array)

那些

>>> my_array[0,:,:]
array([[1, 1, 1, ..., 1, 1, 1],
       [1, 1, 1, ..., 1, 1, 1],
       [1, 1, 1, ..., 1, 1, 1],
       ...,
       [1, 1, 1, ..., 1, 1, 1],
       [1, 1, 1, ..., 1, 1, 1],
       [1, 1, 1, ..., 1, 1, 1]])
>>> my_array[0,:,:].shape
(500, 500)

那些

>>> my_array[0,:,:]
array([[1, 1, 1, ..., 1, 1, 1],
       [1, 1, 1, ..., 1, 1, 1],
       [1, 1, 1, ..., 1, 1, 1],
       ...,
       [1, 1, 1, ..., 1, 1, 1],
       [1, 1, 1, ..., 1, 1, 1],
       [1, 1, 1, ..., 1, 1, 1]])
>>> my_array[0,:,:].shape
(500, 500)

---

也许你的意思是转置：

my_array = np.random.rand(3,3)

[[0.03639883, 0.47393186, 0.25823457],
 [0.3732702 , 0.43692148, 0.13893397],
 [0.11873897, 0.71504391, 0.34635178]]

my_array.T[0]:
[0.03639883, 0.3732702, 0.11873897]
my_array.T[1]:
[0.47393186 , 0.43692148, 0.13893397]
my_array.T[2]:
[0.11873897, 0.71504391, 0.34635178]

---

也许你的意思是转置：

my_array = np.random.rand(3,3)

[[0.03639883, 0.47393186, 0.25823457],
 [0.3732702 , 0.43692148, 0.13893397],
 [0.11873897, 0.71504391, 0.34635178]]

my_array.T[0]:
[0.03639883, 0.3732702, 0.11873897]
my_array.T[1]:
[0.47393186 , 0.43692148, 0.13893397]
my_array.T[2]:
[0.11873897, 0.71504391, 0.34635178]

---

你需要哪一个？你需要哪一个？如果你想保留第一个轴，你可以使用

我的数组[0]

，或者

我的数组[:1]

。如果你想保留第一个轴，你可以使用

我的数组[0]

，或者

我的数组[:1]

。