用数组选择轴形成多维数组

Question

我正在尝试使用另一个数组选择多维数组的子集，例如，如果我有：

a=np.linspace(1,30,30)
a=a.reshape(5,3,2)

我想取子集 [:,0,1]，我可以这样说

a_subset=a[:,0,1]

但是，有没有办法定义一个指定该子集的数组/列表然后减去它？这个想法是做这样的事情：

b=[:,0,1]
a_subset=a[b]

不作为“：”工作的项目不被接受（“SyntaxError：无效语法”）

Answer 1

您可以使用numpy.index_exp (docs) 执行此操作，如下所示：

import numpy as np

a = np.linspace(1, 30, 30)
a = a.reshape(5, 3, 2)

b = np.index_exp[:,0,1]
a_subset = a[b]