Python 为numpy中的每一行选择列

全部,

我有一个形状为（32，2，…）的numpy数组

我想使用32 0和1的数组选择轴1中的第一列或第二列。最终的数组应具有形状（32，…）

我试过：

arr[：，select]

，但最终得到了形状（32，32，…）

---

谢谢

您可以执行

a[np.arange（32），选择]

：

a.shape, a[np.arange(32),select].shape
# ((32, 2, 4), (32, 4))

---

这是正确的。虽然我不知道为什么。。。