共計 338 個字符,預計需要花費 1 分鐘才能閱讀完成。
Python 中可以使用 NumPy 庫來對多維數組進行排序。NumPy 中的 sort 函數可以按照指定的軸對數組進行排序。以下是一個示例:
import numpy as np
# 創建一個多維數組
arr = np.array([[3, 2, 4], [1, 6, 5], [8, 7, 9]])
# 按照第一列進行排序
sorted_arr = arr[arr[:, 0].argsort()]
print(sorted_arr)
輸出結果為:
[[1 6 5]
[3 2 4]
[8 7 9]]
在這個示例中,我們創建了一個 3 ×3 的多維數組 arr,并按照第一列進行排序。使用 arr[:, 0].argsort() 可以獲取按照第一列排序的索引,然后將這些索引應用到數組 arr 上即可得到排序后的數組 sorted_arr。
丸趣 TV 網 – 提供最優質的資源集合!
正文完