python numpy 联合间接排序与插入排序

python numpy 联合间接排序与插入排序

"""
    联合间接排序
    插入排序
"""

import numpy as np

# 联合间接排序
prices = np.array([92, 83, 71, 92, 40, 12, 71])
volumes = np.array([100, 251, 4, 12, 709, 34, 75])
print(volumes)
names = ['Product1', 'Product2', 'Product3', 'Product4', 'Product5', 'Product6', 'Product7']

# 先以价格排序 再以 销量逆序排序
ind = np.lexsort((volumes * -1, prices))  # 返回新的索引
print(ind)
for i in ind:
    print(names[i], end=' ')
print("")
print("*" * 45)

# 插入排序
a = np.array([1, 2, 4, 5, 6, 8, 9])  # 必须是排过序的数组
b = np.array([7, 3])

# 查找插入索引列表
c = np.searchsorted(a, b)
print(c)

# 以 C的 索引将 b 插入a 中
d = np.insert(a, c, b)
print(d)