从python中的数据集中过滤索引

【问题标题】：Filtering indexes from dataset in python从python中的数据集中过滤索引
【发布时间】：2020-08-07 07:19:00
【问题描述】：

我有一个形状为 (2000,74,64) 的数据集。如何从第二维（从 74）过滤索引 2、4、12、14、22、24。如何使用 for 循环来做到这一点？

【问题讨论】：

【解决方案1】：

a = np.array(...) # your array (2000, 74, 64)
b = np.zeros((2000, 6, 64)) # new array shape
j = 0 # iterator

for i in range (0, a.shape[1]):
   if i in {2,4,12,14,22,24}:
      b[:,j,:] = a[:,i,:]
      j += 1

【讨论】：