numpy对某一列倒叙排序,同时对某一列正序排序
时间: 2023-07-20 07:24:44 浏览: 109
NumPy排序的实现
可以使用numpy中的argsort函数来实现对某一列的排序操作。
对某一列倒序排序:
```python
import numpy as np
# 创建一个二维数组
a = np.array([[3, 2, 1], [6, 5, 4]])
# 对第1列进行倒序排序
idx = np.argsort(-a[:, 1])
a = a[idx]
print(a)
```
输出结果为:
```
[[6 5 4]
[3 2 1]]
```
对某一列正序排序:
```python
import numpy as np
# 创建一个二维数组
a = np.array([[3, 2, 1], [6, 5, 4]])
# 对第1列进行正序排序
idx = np.argsort(a[:, 1])
a = a[idx]
print(a)
```
输出结果为:
```
[[3 2 1]
[6 5 4]]
```
其中,`np.argsort()`默认返回的是排序后的索引值,可以通过这些索引值来重新排列原始数组的行。通过在排序函数中传递`-`符号,可以实现倒序排序。
阅读全文