从2个Numpy数组获取前N个值及其坐标

Question

我有一个形状为（851，851）的2D numpy数组“ bigrams”，内部有浮点值。我想从该数组中获取前十个值，并且想要它们的坐标。

我知道np.amax(bigrams)可以返回单个最高值，所以基本上这就是我想要的，但是对于前十名来说。

作为一个numpy-noob，我使用循环编写了一些代码来获取每行的最高值，然后使用np.where()获得了坐标，但是我觉得必须有一种更聪明的方法来解决这个问题。

我有一个形状为（851，851）的2D numpy数组“ bigrams”，内部有浮点值。我想从该数组中获取前十个值，并且想要它们的坐标。我知道np.amax（bigrams）可以返回...

Answer 1

您可以展平并使用argsort。