【数据分析&数据挖掘】数组的排序

import numpy as np

# 创建数组
arr = np.array([[8, 1, 9], [7, 6, 5]])
print("arr: 
", arr)

# sort()直接排序
# 在列的方向上排序
arr.sort(axis=-1)
print("axis=-1排序之后的结果:
", arr)
arr.sort(axis=1)
print("axis=1排序之后的结果:
", arr)
# 在行的方向上排序
arr.sort(axis=0)
print("axis=0排序之后的结果:
", arr)

# argsort()间接排序
arr = np.array([5, 4, 6, 7, 1])
print("arr:
", arr)
# 升序排序之后元素原来所处的下标
res = arr.argsort()

print("res: 
", res)

# 二维数组
arr = np.array([[3, 9, 1], [0, 8, 5]])
print("arr:
", arr)
# 按照列的方向排序
res = arr.argsort(axis=-1)
# 按照行的方向排序
res = arr.argsort(axis=0)

print("res:
", res)

# 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])
print("res: 
", res)

res_a = [a[i] for i in res]
res_b = [b[i] for i in res]
res_c = [c[i] for i in res]

print("a按照c的规则排序之后的结果: 
", res_a)
print("b按照c的规则排序之后的结果: 
", res_b)
print("c按照c的规则排序之后的结果: 
", res_c)