In [62]: a
Out[62]:
array([[1, 2],
[3, 4]])
有没有简单的方法获取[2,3],即第一行的第二个元素和第二行的第一个元素?我有每行的索引列表,即本例中的 [1,0] 。我尝试过[:,[1,0]],但不起作用。
最佳答案
您需要为您想要的所有元素指定 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])
关于python - 选择numpy数组不同行的不同列的元素,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/30113118/