본문 바로가기
파이썬

파이썬 다차원 배열에서 가장 높은 값을 반환하는 방법은 무엇입니까?

by º기록 2021. 1. 21.
반응형

다음과 같은 다차원 배열이 있다고 가정합니다.

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

그러나 이것을 수행하는 더 깨끗한 방법이 있는지 궁금합니다.

*이 경우 최고 = 최소 코드 줄

 

해결 방법

 

주석에서 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])


 

참조 페이지 https://stackoverflow.com/questions/15917076

 

 

반응형

댓글