我有一個 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
])
我如何獲得輸出:
[5, 9, 3, 4]
uj5u.com熱心網友回復:
嘗試這個:
mat[row_col[:, 0], row_col[:, 1]]
輸出:array([5, 9, 3, 4])
轉載請註明出處,本文鏈接:https://www.uj5u.com/houduan/505105.html
標籤:Python python-3.x 麻木的
