1.数组排序:(sort、argsort、lexsort)
import numpy as np
# # 创建一个数组
arr = np.array([[3,2,5,8,1],[9,5,3,1,2]])
# print("arr:\n",arr)
"""
arr:
[[3 2 5 8 1]
[9 5 3 1 2]]
"""
#------------------分割线-------------------------
# # 对数组进行排序 --可以通过axis来指定排序的轴
arr.sort(axis=1)
# print("arr:\n",arr)
"""
arr:
[[1 2 3 5 8]
[1 2 3 5 9]]
"""
#------------------分割线-------------------------
# argsort --返回排序后元素原来的下标
# 创建一个一维数组
arr = np.array([3,4,2,10,5,8,7,6])
res = arr.argsort()
# print(res)
"""
[2 0 1 4 7 6 5 3]
"""
#------------------分割线-------------------------
# 利用列表推导式进行 arr排序结果展示
new_arr = [arr[i] for i in res]
# print("new_arr:\n",new_arr)
"""
new_arr:
[2, 3, 4, 5, 6, 7, 8, 10]
"""
#------------------分割线-------------------------
# lexsort
# 借助传入最后一个数组的排序规则对前面数组进行排序
a = np.array([3, 2, 6, 4, 5])
b = np.array([50, 30, 40, 20, 10])
c = np.array([400, 300, 600, 100, 200])
res = np.lexsort((a, b, c))
# 展示lexsort排序返回值
# print("res:\n",res)
"""
res:
[3 4 1 0 2]
"""
#------------------分割线-------------------------
# 展示排序后的abc 数组
sorted_a = [a[i] for i in res]
sorted_b = [b[i] for i in res]
sorted_c = [c[i] for i in res]
# print("sorted_a:\n",sorted_a)
"""
sorted_a:
[4, 5, 2, 3, 6]
"&