12

假设我有一个多维数组,如下所示:

[
   [.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)

但我很好奇是否有更清洁的方法来做到这一点?

*在这种情况下,最好 = 最少的代码行

4

7 回答 7

13

尝试这个:

max(array.flatten())
于 2013-04-10T05:00:54.307 回答
11

mapwithmax是更清洁的 IMO。

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

地图文档

于 2013-04-10T04:46:07.193 回答
11

由于您在评论中提到您正在使用 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])

编辑: 文档在这里

于 2013-04-10T04:53:14.170 回答
4

使用列表推导:

maxed = [max(sub_array) for sub_array in array]
于 2013-04-10T04:45:00.347 回答
2
 map(max,my_array)

我觉得很短...

于 2013-04-10T04:46:36.183 回答
0

也许不是第二个 for 循环,而是使用max函数

于 2013-04-10T04:48:29.403 回答
0

你可以使用 numpy

import numpy as np
w = [[2,3,4],[5,6,7],[1,2,11]]
h=[]
for i in w:
    h.append(np.amax(i))
print(h)
于 2018-10-25T19:00:12.480 回答