通过使用元组列表作为索引从 numpy 2d 数组中获取值

Question

我有一个元组列表：我有我的2d numpy数组：[(0,0), (1,1), (2,2), (3,3), (4,4)]

array([[8, 6, 5, 9, 3],
       [7, 9, 7, 9, 1],
       [2, 1, 8, 8, 6],
       [7, 1, 5, 1, 3],
       [6, 7, 1, 1, 5]])

如何通过使用带有numpy的列表位置从2d数组中获取值？我应该得到对角线：[8,9,8,1,5]

Answer 1

试试这个，>>> import numpy as np>>> req_index = [(0,0), (1,1), (2,2), (3,3), (4,4)] # this is your tuple index list>>> arr = np.array([[8, 6, 5, 9, 3],        [7, 9, 7, 9, 1],        [2, 1, 8, 8, 6],        [7, 1, 5, 1, 3],        [6, 7, 1, 1, 5]])        >>>输出：>>> [arr[i][j] for i, j in req_index] [8, 9, 8, 1, 5]

Answer 2

您可以转置元组列表，并将这些元组作为项传递到元组中：>>> b = [(0, 0), (1, 1), (2, 2), (3, 3), (4, 4)]>>> a[tuple(np.transpose(b))]array([8, 9, 8, 1, 5])

Answer 3

以下是执行此操作的一种方法： a=np.array([[8, 6, 5, 9, 3],    [7, 9, 7, 9, 1],    [2, 1, 8, 8, 6],    [7, 1, 5, 1, 3],    [6, 7, 1, 1, 5]]) np.diag(a)指纹。array([8, 9, 8, 1, 5])