np array 排序
时间: 2023-10-23 16:02:38 浏览: 92
python 实现多维数组(array)排序
np array 是 NumPy 库中的一种数据结构,用于存储和处理多维数组。要对 np array 进行排序,可以使用 NumPy 库中的 sort 函数。
sort 函数可以按照指定的轴对数组进行排序,也可以按照指定的排序算法进行排序。默认情况下,sort 函数会沿着最后一个轴对数组进行排序。
如果要对一维数组进行排序,只需直接调用 sort 函数即可。例如:
import numpy as np
arr = np.array([3, 1, 4, 2])
sorted_arr = np.sort(arr)
print(sorted_arr)
# 输出:[1 2 3 4]
如果要对二维数组进行排序,可以指定要排序的轴。例如,要按照每一行的第一个元素进行排序:
import numpy as np
arr = np.array([[3, 1, 4, 2], [2, 4, 1, 3]])
sorted_arr = np.sort(arr, axis=1)
print(sorted_arr)
# 输出:[[1 2 3 4]
# [1 2 3 4]]
此外,sort 函数还可以指定排序算法。默认情况下,sort 函数使用快速排序算法。如果要使用其他算法,可以使用 kind 参数进行指定。例如,要使用归并排序算法对数组进行排序:
import numpy as np
arr = np.array([3, 1, 4, 2])
sorted_arr = np.sort(arr, kind='mergesort')
print(sorted_arr)
# 输出:[1 2 3 4]
总之,通过使用 sort 函数,我们可以轻松地对 np array 进行排序,无论是一维数组还是多维数组。
阅读全文