Answers for "get n largest values from 2D numpy array matrix"

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

Python Answers by Framework

Browse Popular Code Answers by Language