在numpy数组的不同行中选择不同列的元素

【问题标题】：select elements of different columns at different rows of numpy array在numpy数组的不同行中选择不同列的元素
【发布时间】：2015-05-07 23:06:49
【问题描述】：

In [62]: a
Out[62]: 
array([[1, 2],
       [3, 4]])

有没有简单的方法来获取[2,3]，即第一行的第二个元素，第二行的第一个元素？我有每行的索引列表，即在这种情况下为 [1,0]。我试过 [:,[1,0]]，但它不起作用。

【问题讨论】：

【解决方案1】：

您需要为所需的所有元素同时指定 i 和 j。例如：

import numpy as np
a = np.array([[1, 2],
              [3, 4]])
i = [0, 1]
j = [1, 0]
print(a[i, j])
# [2, 3]

如果您需要每一行中的一项，您可以使用i = np.arange(a.shape[0])

【讨论】：