У меня есть список:
a = [32, 37, 28, 30, 37, 25, 27, 24, 35, 55, 23, 31, 55, 21, 40, 18, 50,
35, 41, 49, 37, 19, 40, 41, 31]
Максимальный элемент равен 55 (два элемента в позициях 9 и 12)
Мне нужно найти, на каком месте находится (-ы) максимальное значение. Пожалуйста, помогите.
>>> m = max(a)
>>> [i for i, j in enumerate(a) if j == m]
[9, 12]
a.index(max(a))
сообщит вам индекс первого экземпляра самого высокого значения элемента списка a.
Выбранный ответ (и большинство других) требует по крайней мере двух проходов через список.
Здесь однопроходное решение, которое может быть лучшим выбором для более длинных списков.
Отредактировано: Чтобы устранить два недостатка, отмеченные @John Machin. Для (2) я попытался оптимизировать тесты на основе предполагаемой вероятности появления каждого условия и допущений, допущенных от предшественников. Было немного сложно понять правильные значения инициализации для max_val и max_indices, которые работали во всех возможных случаях, особенно если max оказался первым значением в списке, но я считаю, что теперь это делает.
def maxelements(seq):
''' Return list of position(s) of largest element '''
max_indices = []
if seq:
max_val = seq[0]
for i,val in ((i,val) for i,val in enumerate(seq) if val >= max_val):
if val == max_val:
max_indices.append(i)
else:
max_val = val
max_indices = [i]
return max_indices
Я не могу воспроизвести производительность @SilentGhost-beating, указанную @martineau. Здесь мои усилия с сравнениями:
=== maxelements.py ===
a = [32, 37, 28, 30, 37, 25, 27, 24, 35, 55, 23, 31, 55, 21, 40, 18, 50,
35, 41, 49, 37, 19, 40, 41, 31]
b = range(10000)
c = range(10000 - 1, -1, -1)
d = b + c
def maxelements_s(seq): # @SilentGhost
''' Return list of position(s) of largest element '''
m = max(seq)
return [i for i, j in enumerate(seq) if j == m]
def maxelements_m(seq): # @martineau
''' Return list of position(s) of largest element '''
max_indices = []
if len(seq):
max_val = seq[0]
for i, val in ((i, val) for i, val in enumerate(seq) if val >= max_val):
if val == max_val:
max_indices.append(i)
else:
max_val = val
max_indices = [i]
return max_indices
def maxelements_j(seq): # @John Machin
''' Return list of position(s) of largest element '''
if not seq: return []
max_val = seq[0] if seq[0] >= seq[-1] else seq[-1]
max_indices = []
for i, val in enumerate(seq):
if val < max_val: continue
if val == max_val:
max_indices.append(i)
else:
max_val = val
max_indices = [i]
return max_indices
Результаты взломанного старого ноутбука, работающего под управлением Python 2.7 на Windows XP SP3:
>\python27\python -mtimeit -s"import maxelements as me" "me.maxelements_s(me.a)"
100000 loops, best of 3: 6.88 usec per loop
>\python27\python -mtimeit -s"import maxelements as me" "me.maxelements_m(me.a)"
100000 loops, best of 3: 11.1 usec per loop
>\python27\python -mtimeit -s"import maxelements as me" "me.maxelements_j(me.a)"
100000 loops, best of 3: 8.51 usec per loop
>\python27\python -mtimeit -s"import maxelements as me;a100=me.a*100" "me.maxelements_s(a100)"
1000 loops, best of 3: 535 usec per loop
>\python27\python -mtimeit -s"import maxelements as me;a100=me.a*100" "me.maxelements_m(a100)"
1000 loops, best of 3: 558 usec per loop
>\python27\python -mtimeit -s"import maxelements as me;a100=me.a*100" "me.maxelements_j(a100)"
1000 loops, best of 3: 489 usec per loop
Я придумал следующее, и он работает, как вы можете видеть с помощью max, min и других функций над списками, подобными этим:
Итак, рассмотрите следующий список примеров, чтобы узнать положение максимум в списке a:
>>> a = [3,2,1, 4,5]
Используя генератор enumerate и сделав кастинг
>>> list(enumerate(a))
[(0, 3), (1, 2), (2, 1), (3, 4), (4, 5)]
В этот момент мы можем извлечь положение max с помощью
>>> max(enumerate(a), key=(lambda x: x[1]))
(4, 5)
Выше сказано, что максимум находится в позиции 4, а его значение равно 5.
Как вы видите, в аргументе key вы можете найти максимум над любым итерируемым объектом, указав подходящий лямбда.
Я надеюсь, что это вносит свой вклад.
PD: Как заметил @PaulOyster в комментарии. С помощью Python 3.x min и max разрешить новое ключевое слово default, чтобы избежать исключения raise ValueError, когда аргумент пуст. max(enumerate(list), key=(lambda x:x[1]), default = -1)
a = [32, 37, 28, 30, 37, 25, 27, 24, 35,
55, 23, 31, 55, 21, 40, 18, 50,
35, 41, 49, 37, 19, 40, 41, 31]
import pandas as pd
pd.Series(a).idxmax()
9
Вот как я обычно это делаю.
Вы также можете использовать пакет numpy:
import numpy as np
A = np.array(a)
maximum_indices = np.where(A==max(a))
Это вернет массив numpy всех индексов, которые содержат максимальное значение
если вы хотите превратить это в список:
maximum_indices_list = maximum_indices.tolist()
>>> max(enumerate([1,2,3,32,1,5,7,9]),key=lambda x: x[1])
>>> (3, 32)
@shash ответил на это в другом месте
Питоновский способ найти индекс максимального элемента списка будет
position = max(enumerate(a), key=lambda x: x[1])[0]
Который проходит. Тем не менее, он медленнее, чем решение @Silent_Ghost и, тем более, @nmichaels:
for i in s m j n; do echo $i; python -mtimeit -s"import maxelements as me" "me.maxelements_${i}(me.a)"; done
s
100000 loops, best of 3: 3.13 usec per loop
m
100000 loops, best of 3: 4.99 usec per loop
j
100000 loops, best of 3: 3.71 usec per loop
n
1000000 loops, best of 3: 1.31 usec per loop
Вот максимальное значение и индексы, которые он показывает:
>>> from collections import defaultdict
>>> d = defaultdict(list)
>>> a = [32, 37, 28, 30, 37, 25, 27, 24, 35, 55, 23, 31, 55, 21, 40, 18, 50, 35, 41, 49, 37, 19, 40, 41, 31]
>>> for i, x in enumerate(a):
... d[x].append(i)
...
>>> k = max(d.keys())
>>> print k, d[k]
55 [9, 12]
Позже: для удовлетворения @SilentGhost
>>> from itertools import takewhile
>>> import heapq
>>>
>>> def popper(heap):
... while heap:
... yield heapq.heappop(heap)
...
>>> a = [32, 37, 28, 30, 37, 25, 27, 24, 35, 55, 23, 31, 55, 21, 40, 18, 50, 35, 41, 49, 37, 19, 40, 41, 31]
>>> h = [(-x, i) for i, x in enumerate(a)]
>>> heapq.heapify(h)
>>>
>>> largest = heapq.heappop(h)
>>> indexes = [largest[1]] + [x[1] for x in takewhile(lambda large: large[0] == largest[0], popper(h))]
>>> print -largest[0], indexes
55 [9, 12]
Подобная идея со списком, но без перечисления
m = max(a)
[i for i in range(len(a)) if a[i] == m]
Только одна строка:
idx = max(range(len(a)), key = lambda i: a[i])
import operator
def max_positions(iterable, key=None, reverse=False):
if key is None:
def key(x):
return x
if reverse:
better = operator.lt
else:
better = operator.gt
it = enumerate(iterable)
for pos, item in it:
break
else:
raise ValueError("max_positions: empty iterable")
# note this is the same exception type raised by max([])
cur_max = key(item)
cur_pos = [pos]
for pos, item in it:
k = key(item)
if better(k, cur_max):
cur_max = k
cur_pos = [pos]
elif k == cur_max:
cur_pos.append(pos)
return cur_max, cur_pos
def min_positions(iterable, key=None, reverse=False):
return max_positions(iterable, key, not reverse)
>>> L = range(10) * 2
>>> L
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
>>> max_positions(L)
(9, [9, 19])
>>> min_positions(L)
(0, [0, 10])
>>> max_positions(L, key=lambda x: x // 2, reverse=True)
(0, [0, 1, 10, 11])
Этот код не такой сложный, как ответы, опубликованные ранее, но он будет работать:
m = max(a)
n = 0 # frequency of max (a)
for number in a :
if number == m :
n = n + 1
ilist = [None] * n # a list containing index values of maximum number in list a.
ilistindex = 0
aindex = 0 # required index value.
for number in a :
if number == m :
ilist[ilistindex] = aindex
ilistindex = ilistindex + 1
aindex = aindex + 1
print ilist
ilist в приведенном выше коде будет содержать все позиции максимального числа в списке.
Вы можете сделать это различными способами.
Старый традиционный способ:
maxIndexList = list() #this list will store indices of maximum values
maximumValue = max(a) #get maximum value of the list
length = len(a) #calculate length of the array
for i in range(length): #loop through 0 to length-1 (because, 0 based indexing)
if a[i]==maximumValue: #if any value of list a is equal to maximum value then store its index to maxIndexList
maxIndexList.append(i)
print(maxIndexList) #finally print the list
Другой способ без вычисления длины списка и сохранения максимального значения для любой переменной,
maxIndexList = list()
index = 0 #variable to store index
for i in a: #iterate through the list (actually iterating through the value of list, not index )
if i==max(a): #max(a) returns a maximum value of list.
maxIndexList.append(index) #store the index of maximum value
index = index+1 #increment the index
print(maxIndexList)
Мы можем сделать это в Pythonic и умном пути! Используя понимание списка только в одной строке,
maxIndexList = [i for i,j in enumerate(a) if j==max(a)] #here,i=index and j = value of that index
Все мои коды находятся в Python 3.
Если вы хотите получить индексы наибольших чисел n в списке с именем data, вы можете использовать Pandas sort_values:
pd.Series(data).sort_values(ascending=False).index[0:n]
Я знаю, что это очень старый вопрос, но сегодня вы делаете это с
Argmax
https://docs.scipy.org/doc/numpy/reference/generated/numpy.argmax.html
Если вы можете использовать numpy, сделайте np.argmax(a)