Cách nhanh nhất để tìm giá trị lớn nhất thứ n trong một ma trận NumPy

Question

Có rất nhiều giải pháp để làm việc này cho một mảng duy nhất, nhưng những gì về một ma trận, chẳng hạn như:

>>> k 
array([[ 35, 48, 63], 
     [ 60, 77, 96], 
     [ 91, 112, 135]]) 

Bạn có thể sử dụng k.max(), nhưng tất nhiên điều này chỉ trả về giá trị cao nhất, 135. Nếu tôi muốn thứ hai hoặc thứ ba thì sao?

Answer 1

Bạn có thể flatten ma trận và sau đó sắp xếp nó:

>>> k = np.array([[ 35, 48, 63], 
...  [ 60, 77, 96], 
...  [ 91, 112, 135]]) 
>>> flat=k.flatten() 
>>> flat.sort() 
>>> flat 
array([ 35, 48, 60, 63, 77, 91, 96, 112, 135]) 
>>> flat[-2] 
112 
>>> flat[-3] 
96 

Answer 2

Như said, np.partition nên nhanh hơn (tối đa là O (n) thời gian chạy):

np.partition(k.flatten(), -2)[-2] 

nên trả lại 2 yếu tố lớn nhất. (partition đảm bảo rằng phần tử được đánh số ở vị trí, tất cả các phần tử trước đó nhỏ hơn và tất cả các phần tử phía sau đều lớn hơn).

Answer 3

import numpy as np 
a=np.array([[1,2,3],[4,5,6]]) 
a=a.reshape((a.shape[0])*(a.shape[1])) # n is the nth largest taken by us 
print(a[np.argsort()[-n]])