Как вернуть максимальное значение из многомерного массива?

Question

Скажем, у меня есть мульти двумерный массив, как следующие:

[ 
    [.1, .2, .9], 
    [.3, .4, .5], 
    [.2, .4, .8] 
] 

Что бы лучший * способ вернуть единый массив размера, который содержит наибольшее значение из каждой подрешетки ([.9,.5,.8])? Я предполагаю, что я мог бы сделать это вручную, делая что-то вроде:

newArray = [] 
for subarray in array: 
    maxItem = 0 
    for item in subarray: 
     if item > maxItem: 
      maxItem = item 
    newArray.append(maxItem) 

Но мне любопытно, есть ли более чистый способ сделать это?

* В этом случае лучше всего = наименьшее количество строк кода

Answer 1

Поскольку вы упомянули в комментарии, что вы используете NumPy ...

>>> import numpy as np 
>>> a = np.random.rand(3,3) 
>>> a 
array([[ 0.43852835, 0.07928864, 0.33829191], 
     [ 0.60776121, 0.02688291, 0.67274362], 
     [ 0.2188034 , 0.58202254, 0.44704166]]) 
>>> a.max(axis=1) 
array([ 0.43852835, 0.67274362, 0.58202254]) 

---

редактировать: документация here

Answer 2

Использование списка понимание:

maxed = [max(sub_array) for sub_array in array] 

Answer 3

map с max чище ИМО.

>>> arr = [ 
... [.1, .2, .9], 
... [.3, .4, .5], 
... [.2, .4, .8] 
... ] 
>>> map(max, arr) 
[0.9, 0.5, 0.8] 

map documentation.

Answer 4

map(max,my_array) 

Я думаю, что тот довольно короткий ...

Answer 5

Может быть, вместо того, чтобы второй цикл просто использовать max функцию

Answer 6

Попробуйте это:

max(array.flatten())