np.sort函数
时间: 2023-07-19 21:17:50 浏览: 156
Python 列表 sort() 方法
np.sort是numpy库中的一个函数,用于对数组进行排序。它可以按行或列对数组进行排序,并可以指定升序或降序排序。具体用法如下:
```python
import numpy as np
# 对一维数组进行排序
arr = np.array([3, 2, 1])
sorted_arr = np.sort(arr)
print(sorted_arr) # [1 2 3]
# 对二维数组按行排序
arr = np.array([[3, 2, 1], [6, 5, 4]])
sorted_arr = np.sort(arr, axis=1)
print(sorted_arr) # [[1 2 3] [4 5 6]]
# 对二维数组按列排序
arr = np.array([[3, 2, 1], [6, 5, 4]])
sorted_arr = np.sort(arr, axis=0)
print(sorted_arr) # [[3 2 1] [6 5 4]]
# 对数组进行降序排序
arr = np.array([3, 2, 1])
sorted_arr = np.sort(arr)[::-1] # 倒序
print(sorted_arr) # [3 2 1]
```
需要注意的是,np.sort函数返回的是排序后的数组,而不是在原数组上进行排序。如果想在原数组上进行排序,可以使用数组自带的sort方法。
阅读全文