首頁 > 後端開發 > Python教學 > 如何按特定列對 NumPy 陣列進行排序?

如何按特定列對 NumPy 陣列進行排序?

DDD
發布: 2024-12-25 08:56:09
原創
712 人瀏覽過

How Can I Sort a NumPy Array by a Specific Column?

按列對 NumPy 數組進行排序

按特定列對 NumPy 數組進行排序通常是資料操作的必要操作。例如,考慮陣列:

a = array([[9, 2, 3],
           [4, 5, 6],
           [7, 0, 5]])
登入後複製

假設我們要根據第二列中的值對 a 的行重新排序。為了實現這一點,我們可以利用 NumPy 中的 argsort() 函數。

a[:, 1].argsort()
登入後複製

上面的表達式將 argsort() 套用於 a 的第二列,產生與排序值相對應的索引陣列。具體來說,它會傳回:

array([1, 0, 2])
登入後複製

這些索引表示 a 的行應重新排序以按第二列升序排序的順序。結合數組索引,我們可以得到排序後的數組:

a[a[:, 1].argsort()]
登入後複製

該表達式返回:

array([[7, 0, 5],
       [9, 2, 3],
       [4, 5, 6]])
登入後複製

根據需要,a 的行已根據第二列排序.

以上是如何按特定列對 NumPy 陣列進行排序?的詳細內容。更多資訊請關注PHP中文網其他相關文章!

來源:php.cn
本網站聲明
本文內容由網友自願投稿,版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容,請聯絡admin@php.cn
熱門教學
更多>
最新下載
更多>
網站特效
網站源碼
網站素材
前端模板