Esiste un codice a una riga per trovare il valore massimo in una matrice?

Question

per trovare il valore massimo in una matrice di numeri, siamo in grado di codificare 5 linee per risolvere il problema:

ans = matrix[0][0] 
for x in range(len(matrix)): 
    for y in range(len(matrix[0])): 
     ans = max(ans, matrix[x][y]) 
return ans 

C'è una soluzione di una riga per questo problema? Quello che mi è venuta è piuttosto scomoda realtà:

return max(max(matrix, key=max)) 

o

return max(map(max, matrix)) 

Answer 1

È possibile utilizzare generator expression per trovare il massimo nella tua matrice. In questo modo puoi evitare di costruire l'elenco completo degli elementi di matrice in memoria.

maximum = max(max(row) for row in matrix) 

invece di lista come indicato in una precedente risposta here

maximum = max([max(row) for row in matrix]) 

Questo da PEP (sezione rationale):

... molti dei casi d'uso fare non è necessario avere una lista completa creata nella memoria . Invece, devono solo scorrere gli elementi uno a un'ora .

...

espressioni Generator sono particolarmente utili con funzioni come sum(), min(), e Max() che riducono un ingresso iterabile a un singolo valore

...

L'utilità delle espressioni del generatore è notevolmente migliorata se combinata con funzioni di riduzione come sum(), min() e max().

Inoltre, dai un'occhiata a questo post SO: Generator Expressions vs. List Comprehension.

Answer 2

da Matrix, suppongo si intende un 2d-list.

max([max(i) for i in matrix]) 

Answer 3

utilizzando numpy.amax:

import numpy as np 
>>> my_array 
array([[1, 2, 3], 
     [9, 8, 6]]) 
>>> np.amax(my_array) 
9 

Answer 4

È inoltre possibile appiattire la matrice:

from itertools import chain 

flatten = chain.from_iterable 

max(flatten(matrix))