在 NumPy 中按列對數組進行排序
按列對數組進行排序是資料分析和操作中的常見任務。 NumPy 提供了執行此操作的有效方法。
問題:
給定一個具有多列的 NumPy 數組,如何以特定的順序對其行進行排序列?
範例:
考慮以下數組a:
a = np.array([[9, 2, 3], [4, 5, 6], [7, 0, 5]])
我們想要按第二列對a 的行進行排序(第1列,索引從0)可取得:
array([[7, 0, 5], [9, 2, 3], [4, 5, 6]])
答案:
要以第二列對a 排序,請使用以下程式碼:
a[a[:, 1].argsort()]
此表達式實現所需的結果,因為:
因此,這種方法提供了一種方便有效的方法來按任何指定列對 NumPy 數組進行排序。
以上是如何依特定列對 NumPy 陣列的行進行排序?的詳細內容。更多資訊請關注PHP中文網其他相關文章!