np.array排序
时间: 2023-07-01 19:05:50 浏览: 73
你可以使用NumPy中的sort函数来对numpy数组进行排序。sort函数有两个参数,分别是数组和排序方式。默认情况下,sort函数会按升序排列数组元素。
例如,下面的代码将一个随机生成的包含10个元素的一维数组进行排序:
```python
import numpy as np
arr = np.random.randint(0, 10, size=10)
print("Original array:", arr)
sorted_arr = np.sort(arr)
print("Sorted array:", sorted_arr)
```
输出结果:
```
Original array: [4 7 0 6 9 2 9 9 4 4]
Sorted array: [0 2 4 4 4 6 7 9 9 9]
```
如果你想按降序排序,可以指定`kind`参数为`'quicksort'`和`order`参数为`None`,然后将结果数组翻转即可:
```python
import numpy as np
arr = np.random.randint(0, 10, size=10)
print("Original array:", arr)
sorted_arr = np.sort(arr, kind='quicksort', order=None)[::-1]
print("Sorted array in descending order:", sorted_arr)
```
输出结果:
```
Original array: [9 4 9 4 5 2 2 1 2 2]
Sorted array in descending order: [9 9 5 4 4 2 2 2 2 1]
```
阅读全文