Python 如何从具有索引的二维数组中获取值

例如，我有以下二维数组

>>>np.array(((1,2),(3,4),(5,6))) 
>>>array([[1, 2],
          [3, 4],
          [5, 6]])

我想从每列中获取一个元素。例如，我想从第一列获得

3

，从第二列获得

6

。

---

如何使用索引[1,2]实现这一点

1

表示第1列中的第2个元素，

2

表示第2列中的第3个元素

您可以使用所谓的：

---

或者您可以使用：

In [57]: x = np.array(((1,2),(3,4),(5,6)))

# np.arange(x.shape[1]) gives [0,1], the column indices
# needed to select "one from each column"
In [58]: x[[1,2], np.arange(x.shape[1])]
Out[58]: array([3, 6])

In [44]: np.choose([1,2], x)
Out[44]: array([3, 6])