我有一个numpy数组X,我想返回另一个数组Y,其条目是X中n个最大值的索引,即假设我有:
a =np.array[[1, 3, 5], [4, 5 ,6], [9, 1, 7]]
如果我想要前5个"maxs"的索引-这里的最大值为9、7、6、5、5,它们的索引分别是:
b=np.array[[2, 0], [2 2], [ 2 1], [1 1], [0 , 2])
我能找到一些解决方案,使这个方法对于一个一维数组起作用,比如:
c=np.array[1, 2, 3, 4, 5, 6]:
def f(a,N):
return np.argsort(a)[::-1][:N]
但是我们还没有能够在多个维度上生成可用的东西。谢谢!