【Python】np.argsort函数的意义

# np.argsort函数返回的是数组值从小到大的索引值
test_y = np.array([0.2, 1, 0.5, 0.6])
result_y1 = np.argsort(test_y)
print("result_y1:", result_y1) # result_y1: [0 2 3 1]

# 返回的是数组值从大到小的索引值
result_y2 = np.argsort(test_y)[::-1]
print("result_y2:", result_y2) # result_y2: [1 3 2 0]

猜你喜欢

转载自blog.csdn.net/sinat_21791203/article/details/109601685