Numpy按行排序 - Python

当你需要对numpy中的数组按照行进行排序的时候，numpy中提供了很多的排序方法。这里我们将会介绍一些numpy中对于行排序的方法。

按指定行数进行排序

在numpy中，如果你需要按指定的行数对二维数组进行排序，你可以使用以下代码：

import numpy as np

array = np.array([[2,1,4],[1,4,2],[3,2,1]])
sorted_array = array[array[:,1].argsort()]

print(sorted_array)

这里我们使用array[:,1]来表示按照行1进行排序，使用argsort()方法获取到排序后的索引。

根据元素大小进行排序

除了指定行数之外，我们还可以根据元素的大小进行排序。可以使用np.sort()方法或者array.sort()方法。

• 使用np.sort()方法：

import numpy as np

array = np.array([[2,1,4],[1,4,2],[3,2,1]])
sorted_array = np.sort(array)

print(sorted_array)

这里我们使用np.sort()方法进行排序。

• 使用array.sort()方法：

import numpy as np

array = np.array([[2,1,4],[1,4,2],[3,2,1]])
array.sort()

print(array)

这里我们使用array.sort()方法来对数组进行排序。

根据元素大小进行倒序排序

如果我们需要对数组进行倒序排序，可以使用[::-1]来对数组进行反转。

import numpy as np

array = np.array([[2,1,4],[1,4,2],[3,2,1]])
sorted_array = np.sort(array)[::-1]

print(sorted_array)

指定轴数进行排序

如果我们需要指定轴数进行排序，可以使用axis参数来指定。以下是示例代码：

import numpy as np

array = np.array([[2,1,4],[1,4,2],[3,2,1]])
sorted_array = np.sort(array, axis=0)

print(sorted_array)

这里我们使用axis=0来指定按照第一个轴进行排序。

以上就是numpy按行排序的方法介绍。