Фильтрация индексов из набора данных в python

Question

У меня есть набор данных с формой (2000,74,64). Как отфильтровать индексы 2,4,12,14,22,24 из 2-го измерения (из 74). Как я могу это сделать, используя для l oop?

Answer 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