python numpy 二维数组对行排序
时间: 2023-10-11 13:06:33 浏览: 174
二维数组的排序
要对Python中的NumPy二维数组进行行排序,您可以使用NumPy库中的argsort函数和切片操作。下面是一个示例代码:
```python
import numpy as np
# 创建二维数组
arr = np.array([[3, 2, 5], [1, 4, 6], [9, 7, 8]])
# 按行排序
sorted_arr = arr[np.argsort(arr[:, 0])]
# 输出排序后的数组
print(sorted_arr)
```
这将按照第一列的值对二维数组进行升序排序。如果您想要降序排序,只需在argsort函数中添加降序参数即可:
```python
sorted_arr = arr[np.argsort(arr[:, 0])[::-1]]
```
阅读全文