您实际上需要做的是获取数组排序顺序的 反函数 :
import numpy as npx = np.random.rand(10)y = np.empty(x.size,dtype=np.int64)y[x.argsort()] = np.arange(x.size)
运行示例(在ipython中):
In [367]: xOut[367]: array([ 0.09139335, 0.29084225, 0.43560987, 0.92334644, 0.09868977, 0.90202354, 0.80905083, 0.4801967 , 0.99086213, 0.00933582])In [368]: yOut[368]: array([1, 3, 4, 8, 2, 7, 6, 5, 9, 0])
或者,如果要使元素的数量 大于中的 每个对应元素的数量
x,则必须将排序从升序转换为降序。一种可能的选择是简单地交换索引的结构:
y_rev = np.empty(x.size,dtype=np.int64)y_rev[x.argsort()] = np.arange(x.size)[::-1]
@unutbu在评论中建议,另一种方法是将原始数组映射到新数组:
y_rev = x.size - y - 1
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)