argsort函数返回的是数组值从小到大的索引值One dimensional array:一维数组
>>> x = np.array([3, 1, 2])>>> np.argsort(x)array([1, 2, 0])Two-dimensional array:二维数组>>> x = np.array([[0, 3], [2, 2]])>>> xarray([[0, 3],[2, 2]])
>>> np.argsort(x, axis=0) #按列排序array([[0, 1],[1, 0]])>>> np.argsort(x, axis=1) #按行排序array([[0, 1],[0, 1]])>>> np.argsort(x) #按升序排列array([1, 2, 0])>>> np.argsort(-x) #按降序排列array([0, 2, 1])>>> x[np.argsort(x)] #通过索引值排序后的数组array([1, 2, 3])>>> x[np.argsort(-x)]array([3, 2, 1])
转载于:https://www.cnblogs.com/ymjyqsx/p/6425577.html