考慮一個 mxn 矩陣 A。我計算了每一行的最大值的索引,產生了一個維度為 m 的陣列。
如何使用此索引陣列將每行中與 A 形狀相同的第二個矩陣 B 中的值設定為 0。
例子:
A = [[1,2] ,[3,4]]
np.argmax(A,1) --> [1,1]
B = [[1,1] ,[1,1]]
我希望有:
B = [[1 0] ,[1 0]]
我怎樣才能做到這一點?
uj5u.com熱心網友回復:
您可以使用:
B[np.arange(B.shape[0]),np.argmax(A,1)] = 0
轉載請註明出處,本文鏈接:https://www.uj5u.com/gongcheng/412349.html
標籤:
