목록에서 최대 값과 색인을 찾는 Pythonic 방법?


151

목록에서 최대 값을 원하면을 쓸 수 max(List)있지만 최대 값의 색인도 필요한 경우 어떻게해야합니까?

나는 이런 식으로 쓸 수 있습니다 :

maximum=0
for i,value in enumerate(List):
    if value>maximum:
        maximum=value
        index=i

그러나 그것은 지루한 것처럼 보입니다.

그리고 내가 쓰면 :

List.index(max(List))

그런 다음 목록을 두 번 반복합니다.

더 좋은 방법이 있습니까?


"목록을 두 번 통과합니다"는 무슨 뜻입니까? List.index (max (List))는 저에게 효과적입니다.
mwcz 2016 년

14
@ mwc : 최대 값을 결정하기 위해 목록을 한 번 반복 한 다음 해당 값의 색인을 찾기 위해 두 번째로 반복합니다.

10
최대 값이 중복되면 list.index ()가 문제가되지 않습니까?
Logan Yang

@LoganYang 예, 같은 값을 가진 두 개의 항목이있을 수 있습니다.
Florian

순서가 중요하지 않은 경우 List.sort () [-1]
Florian

답변:


186

예를 들어 다음과 같은 많은 옵션이 있습니다.

import operator
index, value = max(enumerate(my_list), key=operator.itemgetter(1))

2
아, 나는 이것을 다른 곳에서 보았지만 튜플이 아닌 하나의 값만 반환 할 것이라고 생각했습니다.
Sunny88

1
@ Sunny88 :이 key함수는 어떤 요소가 최대인지 결정하는 데만 사용됩니다. 요소는 변경되지 않습니다.
Sven Marnach

6
@SvenMarnach 왜 key=lambda e: e[1]대신 수입을 피하지 않습니까?
수명 균형

8
@lifebalance 사용하는 itemgetter()것이 더 빠르며 수입을 피하는 것이 추구 할만한 목표가 아닙니다. 경우에 따라 외부 종속성을 피하는 것이 가치가 있지만 표준 라이브러리에서 가져 오는 것은 중요하지 않습니다.
Sven Marnach '

324

나는 대답이 훌륭하다고 생각하지만 왜 명시 적으로하지 않습니까? 더 많은 사람들이 귀하의 코드를 이해할 것이라고 생각하며 이는 PEP 8과 일치합니다.

max_value = max(my_list)
max_index = my_list.index(max_value)

이 방법은 허용되는 답변보다 약 3 배 빠릅니다.

import random
from datetime import datetime
import operator

def explicit(l):
    max_val = max(l)
    max_idx = l.index(max_val)
    return max_idx, max_val

def implicit(l):
    max_idx, max_val = max(enumerate(l), key=operator.itemgetter(1))
    return max_idx, max_val

if __name__ == "__main__":
    from timeit import Timer
    t = Timer("explicit(l)", "from __main__ import explicit, implicit; "
          "import random; import operator;"
          "l = [random.random() for _ in xrange(100)]")
    print "Explicit: %.2f usec/pass" % (1000000 * t.timeit(number=100000)/100000)

    t = Timer("implicit(l)", "from __main__ import explicit, implicit; "
          "import random; import operator;"
          "l = [random.random() for _ in xrange(100)]")
    print "Implicit: %.2f usec/pass" % (1000000 * t.timeit(number=100000)/100000)

그들이 내 컴퓨터에서 실행 결과 :

Explicit: 8.07 usec/pass
Implicit: 22.86 usec/pass

다른 세트 :

Explicit: 6.80 usec/pass
Implicit: 19.01 usec/pass

3
더 빠를 것으로 기대하지 않았습니다. l을 "xrange (10000000)] + [2]"의 _에 대해 "l = [random.random ()"으로 바꾸어도 더 빠르므로 마지막 요소가 가장 큽니다.
Sunny88

14
@ Sunny88 : 간단한 숫자 목록의 경우 간단한 접근 방식이 더 빠릅니다. 이 경우 성능이 저하되면 numpy.argmax()내 컴퓨터에서 30 배 더 빠른 을 사용 하는 것이 좋습니다 . 목록에 단순한 숫자보다 복잡한 객체가 포함되어 있으면 대답의 접근 방식이 더 빨라질 수 있습니다. 이 방법의 또 다른 장점은 목록뿐만 아니라 임의 반복기에 사용할 수 있다는 것입니다.
Sven Marnach 2016 년

@ Sven-Marnach 먼저 목록을 numpy 배열로 변환해야한다면 numpy가 더 빠를까요? 간단한 예 [0,1,0]에서 더 빠를까요?
tommy.carstensen

1
@ Sven-Marnach 방금 확인했습니다. numpy.argmax는 가장 느린 방법이며 배열에 부동 소수점이나 정수 대신 문자열이 포함되어 있으면 잘못된 대답을 제공합니다.
tommy.carstensen

9
최대 값이 중복되면 list.index ()가 문제가되지 않습니까?
Logan Yang

20

이 답변은 목록이 매우 크다고 가정하고 이미 np.array ()라고 가정 할 때 @Escualo보다 33 배 빠릅니다. 테스트가 100 개가 아닌 10000000 개의 요소를보고 있기 때문에 테스트 실행 횟수를 줄여야했습니다.

import random
from datetime import datetime
import operator
import numpy as np

def explicit(l):
    max_val = max(l)
    max_idx = l.index(max_val)
    return max_idx, max_val

def implicit(l):
    max_idx, max_val = max(enumerate(l), key=operator.itemgetter(1))
    return max_idx, max_val

def npmax(l):
    max_idx = np.argmax(l)
    max_val = l[max_idx]
    return (max_idx, max_val)

if __name__ == "__main__":
    from timeit import Timer

t = Timer("npmax(l)", "from __main__ import explicit, implicit, npmax; "
      "import random; import operator; import numpy as np;"
      "l = np.array([random.random() for _ in xrange(10000000)])")
print "Npmax: %.2f msec/pass" % (1000  * t.timeit(number=10)/10 )

t = Timer("explicit(l)", "from __main__ import explicit, implicit; "
      "import random; import operator;"
      "l = [random.random() for _ in xrange(10000000)]")
print "Explicit: %.2f msec/pass" % (1000  * t.timeit(number=10)/10 )

t = Timer("implicit(l)", "from __main__ import explicit, implicit; "
      "import random; import operator;"
      "l = [random.random() for _ in xrange(10000000)]")
print "Implicit: %.2f msec/pass" % (1000  * t.timeit(number=10)/10 )

내 컴퓨터의 결과 :

Npmax: 8.78 msec/pass
Explicit: 290.01 msec/pass
Implicit: 790.27 msec/pass

분명히하기 : 속도 향상은 numpy C 구현 대 순수 파이썬 때문입니까? 아니면 순수 파이썬을 사용하여 @Escualo의 답변을 개선 할 수있는 방법이 있습니까?
최대

파이썬 3.6을 사용하고 싶다면 "l = np.array ([_ in range in _ in range (10000000)])" ") print (f"Npmax : {(1000 * t. timeit (number = 10) / 10) : 5.2f} msec / pass ")
Piotr Siejda

이것은 2.7에 있었다
portforwardpodcast

1
글쎄, 표준 파이썬 목록을 numpy.argmax처리 할 때까지 속도 가 놀랍습니다 . 그런 다음 속도는 명시 적 버전과 암시 적 버전 사이에 있습니다. 그런 것 같아요 단지 목록을 생성하지 않습니다하지만 거기에 몇 가지 추가 정보를 저장 - 예를 들어, 최소 및 최대 값 (단지 가설)에 대한처럼. np.array
Miroslaw Opoka

18

Python의 내장 라이브러리를 사용하면 매우 쉽습니다.

a = [2, 9, -10, 5, 18, 9] 
max(xrange(len(a)), key = lambda x: a[x])

이 이야기 max목록에서 가장 큰 수를 찾기 위해 [0, 1, 2, ..., len(a)]사용자 정의 함수를 사용하여, lambda x: a[x]그 말한다, 0사실입니다 2, 1실제로 9, 등


Python 3에는 xrange가 없으며 Python 2와 Python 3 모두에서 실행될 코드를 작성하려면 range ()를 사용해야합니다.
Chunde Huang

10
max([(v,i) for i,v in enumerate(my_list)])

튜플 이외의 다른 용도로 사용할 수 있기 때문에 더 좋습니다.
wieczorek1990

이것이 정확히 어떻게 작동합니까? 프로세스를 분석 할 수 있습니까?
clabe45

안녕하세요 @ clabe45, 그것은 튜플 (v, i) 목록에서 my_list를 변환합니다. 여기서 v는 내 목록의 모든 항목이고 i는 해당 인덱스이며 튜플은 최대 값 (및 관련 인덱스도 포함)을 가져옵니다
Luis Sobrecueva

4
고마워, 아마도 대답에 그것을 게시 할 수 있습니까? 최대 값을 계산할 때 max모든 튜플의 첫 번째 항목 ( v)을 어떻게 고려하는 지 알고 있습니까?
clabe45

1
@ clabe45이 회신이 너무 늦을 수도 있지만 지금이 스레드를 방문한 다른 사람 (예 : 저)에게는 여기가 stackoverflow.com/questions/18296755/… 입니다. 이 줄은 아닙니다. "기본적으로 max는 첫 번째 인덱스와 항목을 비교합니다. 첫 번째 인덱스가 같으면 두 번째 인덱스를 비교합니다." 그래서 나는 목록으로 시도해 보았습니다 : l = [1,1,1] 그리고 max ([(v, i) for i, v in enumerate (l)]) 그리고 첫 번째 1이 아니라 마지막 하나 : 결과적으로 (1,2). 나는 그것이 설명되기를 바랍니다 :)
Anupam Jain

10

나는 매우 간단한 방법을 제안합니다.

import numpy as np
l = [10, 22, 8, 8, 11]
print(np.argmax(l))
print(np.argmin(l))

도움이 되길 바랍니다.


4
max([(value,index) for index,value in enumerate(your_list)]) #if maximum value is present more than once in your list then this will return index of the last occurrence

최대 값이 두 번 이상 존재하고 모든 지수를 얻으려면,

max_value = max(your_list)
maxIndexList = [index for index,value in enumerate(your_list) if value==max(your_list)]

1
예. 나는 거의 대답을 게시했지만 목록 이해 한 줄짜리에 동일한 논리가있는 동일한 솔루션을 이미 보았습니다.
WalyKu

2

어쨌든 정렬 된 목록이 필요할 수 있습니까?

이 시도:

your_list = [13, 352, 2553, 0.5, 89, 0.4]
sorted_list = sorted(your_list)
index_of_higher_value = your_list.index(sorted_list[-1])

1. 정렬 시간이 더 복잡합니다. 2. sorted_list색인이 아닌 값을 가지므로 작동하지 않습니다.

1

이 글타래를 되살려 서 미안하지만 내 방법은 추가 할 가치가 있다고 생각했습니다.

이 예제의 목록 이름 'list'

list.sort()
print(list[-1])

목록에서 가장 높은 값을 쉽게 인쇄합니다!

list.sort()ASCII 테이블 의 항목 값을 기준으로 목록을 정렬하므로 목록을 가장 낮은 것부터 가장 높은 것까지 효과적으로 정렬합니다. 그런 다음을 사용하여 목록의 마지막 값 (가장 큰 숫자)을 인쇄하십시오 print(list[-1]).

도움이 되었기를 바랍니다!


5
우리는 이런 식으로 색인을 얻을 수 없습니다
toing_toing

-1

다음은 Python의 내장 함수를 사용하여 질문에 대한 완벽한 솔루션입니다.

# Create the List
numbers = input("Enter the elements of the list. Separate each value with a comma. Do not put a comma at the end.\n").split(",") 

# Convert the elements in the list (treated as strings) to integers
numberL = [int(element) for element in numbers] 

# Loop through the list with a for-loop

for elements in numberL:
    maxEle = max(numberL)
    indexMax = numberL.index(maxEle)

print(maxEle)
print(indexMax)
당사 사이트를 사용함과 동시에 당사의 쿠키 정책개인정보 보호정책을 읽고 이해하였음을 인정하는 것으로 간주합니다.
Licensed under cc by-sa 3.0 with attribution required.