python多维list按照某行或者某列排序

最新推荐文章于 2023-02-16 10:29:27 发布

原创最新推荐文章于 2023-02-16 10:29:27 发布 · 1.2w 阅读

4 ·

CC 4.0 BY-SA版权

python 专栏收录该内容

13 篇文章

订阅专栏

本文通过实例演示了如何使用Python的NumPy库进行数组数据的排序。首先随机生成了一个5行3列的数据矩阵，接着展示了如何获取指定列的排序索引，并依据这些索引重新排列原始数据。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import numpy as np

# 随机生成数据
data = np.random.rand(5, 3)
print(data)

# 取得第一列排序索引
index = np.argsort(data, 0)[:, 1]
print(index)

# 根据索引取数据
print(data[index, :])

[[0.58094702 0.64836198 0.38836734]
[0.27973569 0.60531745 0.6331108 ]
[0.03099137 0.03138515 0.63838206]
[0.126183 0.95057383 0.81691109]
[0.63311877 0.60085732 0.72634485]]

[2 4 1 0 3]

[[0.03099137 0.03138515 0.63838206]
[0.63311877 0.60085732 0.72634485]
[0.27973569 0.60531745 0.6331108 ]
[0.58094702 0.64836198 0.38836734]
[0.126183 0.95057383 0.81691109]]