Maneira pitônica de encontrar o valor máximo e seu índice em uma lista?


151

Se eu quiser o valor máximo em uma lista, posso apenas escrever max(List), mas e se eu também precisar do índice do valor máximo?

Eu posso escrever algo como isto:

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

Mas me parece entediante.

E se eu escrever:

List.index(max(List))

Em seguida, iterará a lista duas vezes.

Existe uma maneira melhor?


O que você quer dizer com "passará a lista duas vezes"? List.index (max (List)) funciona para mim.
Mwcz

14
@mwc: iterará a lista uma vez para determinar o valor máximo e, em seguida, iterará uma segunda vez para encontrar o índice desse valor.

10
List.index () não seria problemático se houver valores máximos duplicados?
Logan Yang

@LoganYang sim, pode haver dois itens com o mesmo valor.
Florian

Se o pedido não for importante, você pode fazer algo como List.sort () [- 1]
Florian

Respostas:


186

Existem muitas opções, por exemplo:

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

2
Ah, eu já vi isso em outros lugares, mas achei que ele retornaria apenas um valor, não uma tupla.
Sunny88

1
@ Sunny88: A keyfunção é usada apenas para decidir qual elemento é máximo. Os elementos não são alterados.
Sven Marnach

6
@SvenMarnach Por que não key=lambda e: e[1]evitar a importação?
lifebalance

8
@lifebalance O uso itemgetter()é mais rápido e evitar uma importação não é uma meta que vale a pena perseguir. Evitar dependências externas pode valer a pena em alguns casos, mas uma importação da biblioteca padrão não é um problema.
Sven Marnach 7/08

324

Eu acho que a resposta aceita é ótima, mas por que você não faz isso explicitamente? Sinto que mais pessoas entenderiam seu código, e isso está de acordo com o PEP 8:

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

Este método também é cerca de três vezes mais rápido que a resposta aceita:

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)

Resultados à medida que são executados no meu computador:

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

Outro conjunto:

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

3
Não esperava que fosse mais rápido. É mais rápido mesmo quando eu substituo l por "l = [random.random () para _ in xrange (10000000)] + [2]", o que garante que o último elemento seja o maior.
precisa saber é o seguinte

14
@ Sunny88: Para uma lista simples de números, a abordagem simples é mais rápida. Se você estiver buscando desempenho nesse caso, sugiro que seja usado numpy.argmax()30 vezes mais rápido na minha máquina. Se a lista contiver objetos mais complicados do que meros números, a abordagem em minha resposta pode se tornar mais rápida. Outra vantagem dessa abordagem é que ela pode ser usada para iteradores arbitrários, não apenas para listas.
precisa saber é o seguinte

@ Sven-Marnach O numpy seria mais rápido se eu tivesse que converter minha lista em um array numpy primeiro? Seria mais rápido para o exemplo simples [0,1,0]?
tommy.carstensen

1
@ Sven-Marnach Acabei de verificar. numpy.argmax é de longe o método mais lento, e fornece a resposta errada, se a matriz contiver cadeias de caracteres em vez de números flutuantes ou números inteiros.
tommy.carstensen

9
List.index () não seria problemático se houver valores máximos duplicados?
Logan Yang

20

Esta resposta é 33 vezes mais rápida que o @Escualo, assumindo que a lista é muito grande e assumindo que já é um np.array (). Eu tive que diminuir o número de execuções de teste porque o teste está analisando 10000000 elementos e não apenas 100.

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 )

Resultados no meu computador:

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

Só para esclarecer: a aceleração é apenas devido à implementação numpy de C versus python puro? Ou existe uma maneira de obter alguma melhoria na resposta do @ Escualo usando python puro?
máximo

Se alguém quiser usar o python 3.6, pode fazer algo como: "l = np.array ([random.random () para _ no intervalo (10000000)])") ") print (f" Npmax: {(1000 * t. timeit (número = 10) / 10): 5.2f} mseg / passe ")
Piotr Siejda

Isso foi em 2,7
portforwardpodcast

1
Bem, a velocidade numpy.argmaxparece incrível até você deixar processar uma lista python padrão . Então a velocidade fica entre a versão explícita e a implícita. Eu acho np.arrayque não apenas cria uma lista, mas salva algumas informações extras, como por exemplo, valores mínimo e máximo (apenas uma hipótese).
Miroslaw Opoka 28/03/19

18

Com a biblioteca interna do Python, é bem fácil:

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

Isso indica maxpara encontrar o maior número da lista [0, 1, 2, ..., len(a)], usando a função personalizada lambda x: a[x], que diz que 0é realmente 2, 1é realmente 9etc.


No Python 3, não há xrange; se você deseja escrever código que será executado no Python 2 e no Python 3, use range ().
Chunde Huang

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

Isso é melhor porque você pode adaptá-lo para usar com algo diferente de tupla.
wieczorek1990

Como, exatamente, isso funciona? Você pode quebrar o processo?
clabe45

Oi @ clabe45, converte my_list em uma lista de tuplas (v, i) em que v é cada item da minha lista e i é o índice correspondente, em seguida, obtém a tupla com o valor máximo (e também com o índice associado)
Luis Sobrecueva

4
Obrigado, você pode postar isso na resposta possivelmente? E como maxsaber apenas levar em consideração o primeiro item de cada tupla ( v) ao calcular o valor máximo?
clabe45

1
@ clabe45 Pode ser que essa resposta seja tarde demais, mas para outras pessoas (como eu) que se depararam com esse segmento agora, aqui: stackoverflow.com/questions/18296755/… é uma explicação. Não esta linha: "Por padrão, max irá comparar os itens pelo primeiro índice, se o primeiro índice for o mesmo, ele comparará o segundo índice." Então, tentei com a lista: l = [1,1,1] e, em seguida, max ([(v, i) para i, v no enumerar (l)]) e isso não me dá o primeiro 1, mas o último um: (1,2) como resultado. Espero que explica :)
Anupam Jain

10

Eu sugeriria uma maneira muito simples:

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

Espero que ajude.


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

Se o valor máximo estiver presente mais de uma vez e você desejar obter todos os índices,

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

1
Sim. Eu quase postei uma resposta, mas depois vi que você já tinha a mesma solução com a mesma lógica em uma linha de compreensão de lista.
WalyKu 16/02

2

Talvez você precise de uma lista ordenada, afinal?

Tente o seguinte:

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. A classificação tem maior complexidade de tempo. 2. sorted_listnão possui índices, mas valores, portanto não funcionaria.

1

desculpe por reviver esse tópico, mas achei que meu método valia a pena adicionar.

O nome da lista neste exemplo 'list'

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

Isso imprimirá o valor mais alto da lista com facilidade!

list.sort()classifica a lista pelo valor do item na tabela ASCII , classificando de maneira eficaz a lista do menor para o maior. Em seguida, imprimo o último valor na lista (que será o maior número) usandoprint(list[-1]) .

Espero que isto ajude!


5
não podemos obter o índice dessa maneira
toing_toing 03/07/19

-1

Aqui está uma solução completa para sua pergunta usando as funções internas do 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)
Ao utilizar nosso site, você reconhece que leu e compreendeu nossa Política de Cookies e nossa Política de Privacidade.
Licensed under cc by-sa 3.0 with attribution required.