基於另一個數組從 numpy 數組的行和列中取值

Question

我有一個 np 權重數組

mat = np.array([
    [1, 2, 3],
    [4, 5, 6],
    [7, 8, 9]
])

我有另一個 numpy 數組，其中包含要從權重矩陣中提取的行和索引。

row_col = np.array([
    [1, 1], # row 1, col 1
    [2, 2], # row 2, col 2
    [0, 2], # row 0, col 2
    [1, 0]  # row 1, col 0
])

如何獲得 output：

[5, 9, 3, 4] 

Answer 1

嘗試這個：

mat[row_col[:, 0], row_col[:, 1]]

Output： array([5, 9, 3, 4])