Answers for "take five largest values of numpy array"

0

numpy get index of n largest values

numbers = np.array([1, 3, 2, 4])
n = 2
indices = (-numbers).argsort()[:n]
Posted by: Guest on March-05-2021
0

get n largest values from 2D numpy array matrix

""" for this matrix:
[[ 8,  9,  4],
 [10,  5, 18],
 [ 5,  6,  6]]
"""
n = 5
rows = mat4.shape[0]
row_col_idxs = [(e//rows, e-(e//rows * rows)) 
				for e in mat4.flatten().argsort()[::-1][:n]]
""" result: [(1, 2), (1, 0), (0, 1), (0, 0), (2, 2)] """
Posted by: Guest on September-13-2021

Code answers related to "take five largest values of numpy array"

Python Answers by Framework

Browse Popular Code Answers by Language