温馨提示×

python多维数组排序的方法是什么

python

小亿

169

2023-10-12 18:29:11

栏目: 编程语言

Python开发者专用服务器限时活动，0元免费领，库存有限，领完即止！点击查看>>

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]]

在这个示例中，我们创建了一个3x3的多维数组arr，并按照第一列进行排序。使用arr[:, 0].argsort()可以获取按照第一列排序的索引，然后将这些索引应用到数组arr上即可得到排序后的数组sorted_arr。

0 赞

0 踩

最新问答