子集3d numpy数组

Question

我已经检查了numpy文档，但是某些索引仍然无法理解。 我有一个numpy数组，其形状为（40000，432），其外观类似于：

arr = [[1,2,3......431,432],
       [1,2,3......431,432],
       [1,2,3......431,432],
       ....................
       [1,2,3......431,432]'
       [1,2,3......431,432]]

我想在一个范围内（即20-50）对每个数组进行子集化，以便形状为（40000，30），并且看起来像：

subarr = [[20,21,22...48,49,50],
          [20,21,22...48,49,50],
          [20,21,22...48,49,50],
          .....................
          [20,21,22...48,49,50]]

我尝试的所有操作都会使我出错，或者给我形状（30、432），这不是我所需要的。 如何沿我想要的轴子集2d数组？

Answer 1

您想使用numpy 切片 ：

arr = np.zeros((40000, 432))
subarr = arr[:, 20:50]
print(subarr.shape)

产量

(40000L, 30L)

形状输出中的L表示整数是Python类型的long 。