numpy 数组排序（参考）

import numpy as np


# 创建数组
# arr = np.arange(16, 0, -1)
# print(arr)
# arr.sort()
# print(arr)

# 创建二位数组
# arr = np.arange(16, 0,-1).reshape(4, 4)
# print(arr)
# axis=1按照横向进行排序，axis=0按照纵向进行排序
# arr.sort(axis=1)
# arr.sort(axis=0)
# print(arr)

# 创建一个数组
# arr = np.arange(4, 0, -1)
# print('arr:', arr)
# res = arr.argsort() # 按排序后顺序返回元素排序前的下标
# print(res)
"""
arr: [4 3 2 1]
[3 2 1 0]
[1 2 0]
"""

# 创建三个数组
# arr1 = np.array([3, 2, 5, 7, 0])
# arr2 = np.array([90, 78, 89, 24, 57])
# arr3 = np.array([300, 100, 500, 700, 200])
# res = np.lexsort((arr1, arr2, arr3))
# print(res)
# print(arr1)

surnames = ('Hertz', 'Galilei', 'Hertz')
first_names = ('Heinrich', 'Galileo', 'Gustav')
ind = np.lexsort((first_names, surnames))
print(ind)
"""
[1 2 0]
"""