我有一個(gè)a像這樣的numpy 數(shù)組In [318]: aOut[318]:array([[0. , 1. , 2. , 3. ], [0.5, 0.3, 0.2, 0.25]])我需要沿第二行（帶有 [0.5,0.3,0.2,0.25] 的行）排序，同時(shí)使第一行相應(yīng)更改。在這種情況下，預(yù)期結(jié)果是2 3 1. 00.2, 0.25, 0.3, 0.5我怎樣才能做到這一點(diǎn)？謝謝。我嘗試使用 axis=-1 和 0 進(jìn)行 np.sort; 它們不是我需要的。重要提示：性能是我解決問題的關(guān)鍵。我的圖像處理應(yīng)用程序的數(shù)組通常有 N 列，其中 N 接近 400 萬。
1 回答
四季花海
TA貢獻(xiàn)1811條經(jīng)驗(yàn) 獲得超5個(gè)贊
用于np.argsort()獲取已排序行的索引，然后將其用作掩碼對整個(gè)數(shù)組進(jìn)行排序（基于列）：
In [69]: mask = np.argsort(a[1])
In [70]: a[:, mask]
Out[70]:
array([[2. , 3. , 1. , 0. ],
[0.2 , 0.25, 0.3 , 0.5 ]])
添加回答
舉報(bào)
0/150
提交
取消