Numpy常用函数

  1. np.argsort对数组排序

argsort 返回一个array,此array的每个元素均为非负整数,此array的第i个位置的元素的值表示待排序array的排第i个位置的index(升序排列)。

x = np.array([4, 5, 2, 1, 9, 6])
index = np.argsort(x)
x[index[0]]
# 1
x[index[1]]
# 2
x[index[len(x)-1]]
# 9
# index[i]表示x中按升序排第i个位置的元素的index
# x[index[i]]表示x中按升序排第i个位置的元素的值

你可能感兴趣的:(python编程)