python numpy 二维数组对行排序
时间: 2023-10-11 07:06:33 浏览: 89
要对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]]
```
相关问题
python numpy三维数组排序
你可以使用 numpy 的 sort 函数进行三维数组排序。按照某个维度排序只需要设置 axis 参数即可。以下是一个示例代码,其中我们对一个 3x3x3 的三维数组按照第二维进行排序:
```python
import numpy as np
arr = np.array([[[9, 8, 7], [6, 5, 4], [3, 2, 1]],
[[12, 11, 10], [15, 14, 13], [18, 17, 16]],
[[21, 20, 19], [24, 23, 22], [27, 26, 25]]])
sorted_arr = np.sort(arr, axis=1)
print(sorted_arr)
```
输出结果:
```
[[[ 3 2 1]
[ 6 5 4]
[ 9 8 7]]
[[12 11 10]
[15 14 13]
[18 17 16]]
[[21 20 19]
[24 23 22]
[27 26 25]]]
```
如果你想按照其他维度排序,只需要改变 axis 参数即可。例如按照第三维排序:
```python
import numpy as np
arr = np.array([[[9, 8, 7], [6, 5, 4], [3, 2, 1]],
[[12, 11, 10], [15, 14, 13], [18, 17, 16]],
[[21, 20, 19], [24, 23, 22], [27, 26, 25]]])
sorted_arr = np.sort(arr, axis=2)
print(sorted_arr)
```
输出结果:
```
[[[ 1 2 3]
[ 4 5 6]
[ 7 8 9]]
[[10 11 12]
[13 14 15]
[16 17 18]]
[[19 20 21]
[22 23 24]
[25 26 27]]]
```
如果你想获取排序后的索引,可以使用 argsort 函数,例如:
```python
import numpy as np
arr = np.array([[[9, 8, 7], [6, 5, 4], [3, 2, 1]],
[[12, 11, 10], [15, 14, 13], [18, 17, 16]],
[[21, 20, 19], [24, 23, 22], [27, 26, 25]]])
sorted_indices = np.argsort(arr, axis=2)
print(sorted_indices)
```
输出结果:
```
[[[2 1 0]
[2 1 0]
[2 1 0]]
[[2 1 0]
[2 1 0]
[2 1 0]]
[[2 1 0]
[2 1 0]
[2 1 0]]]
```
python如何对二维数组进行排序
Python中可以使用numpy模块的lexsort方法对二维数组进行排序。lexsort支持对数组按指定行或列的顺序排序,是间接排序,不修改原数组,返回索引。默认按最后一行元素有小到大排序,返回最后一行元素排序后索引所在位置。对于一维数组,a.argsort()这么使用就可以;argsort也不修改原数组,返回索引。具体操作可以参考以下代码:
import numpy as np
# 定义一个二维数组
a = np.array([[2, 7, 4, 2],
[35, 9, 1, 5],
[22, 12, 3, 2]])
# 按最后一列顺序排序
a[np.lexsort(a.T)]
# 按最后一列逆序排序
a[np.lexsort(-a.T)]
# 按第一列顺序排序
a[np.lexsort(a[:, ::-1].T)]
# 按最后一行顺序排序
a.T[np.lexsort(a)].T
# 按第一行顺序排序
a.T[np.lexsort(a[::-1, :])].T
以上代码分别演示了按最后一列顺序排序、按最后一列逆序排序、按第一列顺序排序、按最后一行顺序排序、按第一行顺序排序的操作方法。