Python中numpy使用sort函数对数组进行排序

NumPy中的sort()函数可以用来对数组进行排序。它的用法如下：

import numpy as np

# 创建一个一维数组
arr = np.array([3, 1, 4, 2, 5])

# 对数组进行排序
sorted_arr = np.sort(arr)

print(sorted_arr)

输出结果为：

[1 2 3 4 5]

在这个例子中，我们首先创建了一个一维数组arr，然后使用np.sort()函数对其进行排序，并将结果存储在变量sorted_arr中。最后，我们将结果打印出来。

sort()函数还可以用于对多维数组进行排序。默认情况下，它会沿着最后一个轴进行排序。例如：

# 创建一个二维数组
arr = np.array([[3, 1, 4], [2, 5, 7]])

# 沿着最后一个轴对数组进行排序
sorted_arr = np.sort(arr)

print(sorted_arr)

输出结果为：

[[1 3 4]
 [2 5 7]]

在这个例子中，我们创建了一个二维数组arr，然后使用np.sort()函数对其进行排序。由于arr是一个二维数组，因此np.sort()函数默认会沿着最后一个轴（即列轴）进行排序。所以输出结果中的每一列都是按升序排列的。如果要按照行进行排序，则可以将axis参数设置为1，例如：

# 沿着第一个轴对数组进行排序
sorted_arr = np.sort(arr, axis=1)

print(sorted_arr)

输出结果为：

lua [[1 3 4] [2 5 7]]
在这个例子中，我们将axis参数设置为1，使np.sort()函数沿着第一个轴（即行轴）进行排序。因此，输出结果中的每一行都是按升序排列的。