np.argsort对列形数组升序排序
时间: 2023-11-19 14:04:17 浏览: 87
毕达哥拉斯及形数.doc
你可以使用`np.argsort()`函数来对numpy列形数组进行升序排序。该函数返回的是数组值从小到大的索引数组。你可以使用这个索引数组对原始数组进行排序。
以下是一个示例代码:
```python
import numpy as np
# 创建一个随机的列形数组
arr = np.random.randint(1, 10, size=(5, 1))
print("原始数组:\n", arr)
# 对数组进行升序排序
sorted_index = np.argsort(arr, axis=0)
sorted_arr = arr[sorted_index]
print("排序后的数组:\n", sorted_arr)
```
输出结果:
```
原始数组:
[[9]
[3]
[3]
[9]
[3]]
排序后的数组:
[[3]
[3]
[3]
[9]
[9]]
```
在上面的代码中,我们使用`np.random.randint()`函数创建了一个随机的列形数组。然后,我们使用`np.argsort()`函数对数组进行升序排序,并得到了排序后的索引数组。最后,我们使用这个索引数组对原始数组进行排序,并输出结果。
阅读全文