Pythonowy sposób na znalezienie maksymalnej wartości i jej indeksu na liście?

151

Jeśli chcę mieć maksymalną wartość na liście, mogę po prostu napisać max(List), ale co, jeśli potrzebuję również indeksu maksymalnej wartości?

Mogę napisać coś takiego:

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

Ale wydaje mi się to nudne.

A jeśli napiszę:

List.index(max(List))

Następnie dwukrotnie iteruje listę.

Czy jest lepszy sposób?

Sunny88
źródło
Co masz na myśli mówiąc „przejdzie listę dwa razy”? List.index (max (List)) działa dla mnie.
mwcz
14
@mwc: iteruje listę raz, aby określić maksymalną wartość, a następnie iteruje ją po raz drugi, aby znaleźć indeks tej wartości.
10
Czy metoda list.index () nie byłaby problematyczna, gdyby były zduplikowane wartości maksymalne?
Logan Yang
@LoganYang tak, mogą istnieć dwa elementy o tej samej wartości.
Florian
Jeśli kolejność nie jest ważna, możesz zrobić coś takiego jak List.sort () [- 1]
Florian

Odpowiedzi:

186

Opcji jest wiele, na przykład:

import operator
index, value = max(enumerate(my_list), key=operator.itemgetter(1))
Sven Marnach
źródło
2
Ach, widziałem to w innych miejscach, ale pomyślałem, że zwróci tylko jedną wartość, a nie krotkę.
Sunny88
1
@ Sunny88: keyFunkcja służy tylko do decydowania, który element jest maksymalny. Elementy nie ulegają zmianie.
Sven Marnach
6
@SvenMarnach Dlaczego key=lambda e: e[1]zamiast tego nie uniknąć importu?
saldo ratunkowe
8
@lifebalance Używanie itemgetter()jest szybsze, a unikanie importu nie jest celem, do którego warto dążyć. W niektórych przypadkach warto unikać zewnętrznych zależności, ale import z biblioteki standardowej nie stanowi problemu.
Sven Marnach
324

Myślę, że zaakceptowana odpowiedź jest świetna, ale dlaczego nie zrobisz tego wprost? Wydaje mi się, że więcej ludzi zrozumie Twój kod i jest to zgodne z PEP 8:

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

Ta metoda jest również około trzy razy szybsza niż zaakceptowana odpowiedź:

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)

Wyniki wyświetlane na moim komputerze:

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

Inny zestaw:

Explicit: 6.80 usec/pass
Implicit: 19.01 usec/pass
Escualo
źródło
3
Nie spodziewałem się, że będzie szybszy. Jest szybszy nawet wtedy, gdy zamieniam l na "l = [random.random () for _ in xrange (10000000)] + [2]", co gwarantuje, że ostatni element jest największy.
Sunny88
14
@ Sunny88: W przypadku prostej listy liczb proste podejście jest szybsze. Jeśli zależy ci na wydajności w tym przypadku, proponuję użyć numpy.argmax(), co jest kolejne 30 razy szybsze na moim komputerze. Jeśli lista zawiera bardziej skomplikowane obiekty niż same liczby, podejście w mojej odpowiedzi może stać się szybsze. Inną zaletą tego podejścia jest to, że można go używać do dowolnych iteratorów, a nie tylko do list.
Sven Marnach
@ Sven-Marnach Czy numpy byłby szybszy, gdybym musiał najpierw przekonwertować moją listę na tablicę numpy? Czy byłoby szybsze dla prostego przykładu [0,1,0]?
tommy.carstensen
1
@ Sven-Marnach Właśnie sprawdziłem. numpy.argmax jest zdecydowanie najwolniejszą metodą i daje złą odpowiedź, jeśli tablica zawiera ciągi znaków zamiast liczb zmiennoprzecinkowych lub liczb całkowitych.
tommy.carstensen
9
Czy metoda list.index () nie byłaby problematyczna, gdyby były zduplikowane wartości maksymalne?
Logan Yang
20

Ta odpowiedź jest 33 razy szybsza niż @Escualo przy założeniu, że lista jest bardzo duża i przy założeniu, że jest to już np.array (). Musiałem zmniejszyć liczbę przebiegów testowych, ponieważ test dotyczy 10000000 elementów, a nie tylko 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 )

Wyniki na moim komputerze:

Npmax: 8.78 msec/pass
Explicit: 290.01 msec/pass
Implicit: 790.27 msec/pass
portforwardpodcast
źródło
Dla wyjaśnienia: przyspieszenie wynika po prostu z implementacji numpy C w porównaniu z czystym Pythonem? Czy jest sposób na poprawę odpowiedzi @ Escualo przy użyciu czystego Pythona?
maksymalnie
Jeśli ktoś chciałby używać Pythona 3.6, można zrobić coś takiego: "l = np.array ([random.random () for _ in range (10000000)])") print (f "Npmax: {(1000 * t. timeit (liczba = 10) / 10): 5,2f} msec / pass ")
Piotr Siejda
Było to 2.7
portforwardpodcast,
1
Cóż, szybkość numpy.argmaxwygląda niesamowicie, dopóki nie pozwolisz mu przetworzyć standardowej listy Pythona. Wtedy prędkość leży między wersją jawną i niejawną. Wydaje mi się, że np.arraynie tylko tworzy listę, ale zapisuje w niej dodatkowe informacje - na przykład wartości minimalne i maksymalne (tylko hipoteza).
Mirosław Opoka
18

Dzięki wbudowanej bibliotece Pythona jest to całkiem proste:

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

Mówi to, maxaby znaleźć największą liczbę na liście [0, 1, 2, ..., len(a)], używając funkcji niestandardowej lambda x: a[x], która mówi, że 0tak jest 2, 1jest w rzeczywistości 9itd.

Sunil Kapil
źródło
W Pythonie 3 nie ma xrange, jeśli chcesz napisać kod, który będzie działał zarówno dla Pythona 2, jak i Pythona 3, powinieneś użyć range ().
Chunde Huang
10
max([(v,i) for i,v in enumerate(my_list)])
Luis Sobrecueva
źródło
Jest to lepsze, ponieważ możesz go dostosować do użycia z czymś innym niż krotka.
wieczorek1990
Jak dokładnie to działa? Czy możesz przerwać ten proces?
clabe45
Cześć @ clabe45, konwertuje my_list na listę krotek (v, i), gdzie v jest każdą pozycją z mojej listy, a i jest indeksem korespondującym, a następnie pobiera krotkę z wartością maksymalnego (i również z powiązanym indeksem)
Luis Sobrecueva
4
Dzięki, czy możesz zamieścić to w odpowiedzi? A skąd maxwiadomo, że vprzy obliczaniu maksymalnej wartości należy wziąć pod uwagę po prostu pierwszy element każdej krotki ( )?
clabe45
1
@ clabe45 Może ta odpowiedź przychodzi za późno, ale dla innych (takich jak ja), którzy natknęli się teraz na ten wątek, tutaj: stackoverflow.com/questions/18296755/ ... to wyjaśnienie. Nie w tym wierszu: „Domyślnie max porówna pozycje według pierwszego indeksu, jeśli pierwszy indeks jest taki sam, to porówna drugi indeks”. Więc spróbowałem z listą: l = [1,1,1], a następnie max ([(v, i) for i, v in enumerate (l)]) i daje mi nie pierwszą 1, ale ostatnią one: (1,2) jako wynik. Mam nadzieję, że to wyjaśnia :)
Anupam Jain
10

Sugerowałbym bardzo prosty sposób:

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

Mam nadzieję, że to pomoże.

Igor Manzhos
źródło
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

Jeśli maksymalna wartość występuje więcej niż raz i chcesz uzyskać wszystkie wskaźniki,

max_value = max(your_list)
maxIndexList = [index for index,value in enumerate(your_list) if value==max(your_list)]
Taohidul Islam
źródło
1
Tak. Prawie opublikowałem odpowiedź, ale potem zobaczyłem, że masz już to samo rozwiązanie z tą samą logiką w jednej linijce rozumienia listy.
WalyKu,
2

Może i tak potrzebujesz posortowanej listy?

Spróbuj tego:

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])
Mattias
źródło
1. Sortowanie ma większą złożoność czasową. 2. sorted_listnie ma indeksów, ale wartości, więc nie zadziała.
1

przepraszam za ożywienie tego wątku, ale pomyślałem, że warto dodać moją metodę.

Nazwa listy w tym przykładzie „lista”

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

To spowoduje wydrukowanie największej wartości na liście, tak łatwo jak!

list.sort()sortuje listę według wartości pozycji w tabeli ASCII , więc efektywnie sortuje listę od najniższej do najwyższej. Następnie po prostu drukuję ostatnią wartość na liście (która będzie największą liczbą) za pomocą print(list[-1]).

Mam nadzieję że to pomoże!

Mackey Johnstone
źródło
5
nie możemy uzyskać indeksu w ten sposób
toing_toing
-1

Oto kompletne rozwiązanie twojego pytania przy użyciu wbudowanych funkcji Pythona:

# 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)
Samdom dla pokoju
źródło