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

52次阅读
没有评论

共计 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 网 – 提供最优质的资源集合!

正文完
 
丸趣
版权声明:本站原创文章,由 丸趣 2023-12-21发表,共计338字。
转载说明:除特殊说明外本站除技术相关以外文章皆由网络搜集发布,转载请注明出处。
评论(没有评论)