index()
просто даст первое вхождение элемента в список. Есть ли опрятный трюк, который возвращает все индексы в списке?
Как найти все вхождения элемента в списке?
Ответ 1
Вы можете использовать понимание списка:
indices = [i for i, x in enumerate(my_list) if x == "whatever"]
Ответ 2
Пока это не решение для списков напрямую, numpy
действительно сияет для такого рода вещей:
import numpy as np
values = np.array([1,2,3,1,2,4,5,6,3,2,1])
searchval = 3
ii = np.where(values == searchval)[0]
возвращает:
ii ==>array([2, 8])
Это может быть значительно быстрее для списков (массивов) с большим количеством элементов по сравнению с некоторыми другими решениями.
Ответ 3
Решение с использованием list.index
:
def indices(lst, element):
result = []
offset = -1
while True:
try:
offset = lst.index(element, offset+1)
except ValueError:
return result
result.append(offset)
Это намного быстрее, чем понимание списка с помощью enumerate
, для больших списков. Он также намного медленнее, чем решение numpy
, если у вас уже есть массив, в противном случае стоимость преобразования перевешивает коэффициент усиления (тестируется на целых списках с 100, 1000 и 10000 элементами).
ПРИМЕЧАНИЕ.. Предупреждение, основанное на комментарии Chris_Rands: это решение быстрее, чем понимание списка, если результаты достаточно разрежены, но если в списке есть много экземпляров элемента, который (более 15% от списка, при тестировании со списком из 1000 целых чисел), понимание списка выполняется быстрее.
Ответ 4
Как насчет:
In [1]: l=[1,2,3,4,3,2,5,6,7]
In [2]: [i for i,val in enumerate(l) if val==3]
Out[2]: [2, 4]
Ответ 5
occurrences = lambda s, lst: (i for i,e in enumerate(lst) if e == s)
list(occurrences(1, [1,2,3,1])) # = [0, 3]
Ответ 6
Еще одно решение (извините, если дубликаты) для всех вхождений:
values = [1,2,3,1,2,4,5,6,3,2,1]
map(lambda val: (val, [i for i in xrange(len(values)) if values[i] == val]), values)
Ответ 7
more_itertools.locate
находит индексы для всех элементов, удовлетворяющих условию.
from more_itertools import locate
list(locate([0, 1, 1, 0, 1, 0, 0]))
# [1, 2, 4]
list(locate(['a', 'b', 'c', 'b'], lambda x: x == 'b'))
# [1, 3]
more_itertools
- это сторонняя библиотека > pip install more_itertools
.
Ответ 8
Или Используйте range
(python 3):
l=[i for i in range(len(lst)) if lst[i]=='something...']
Для (python 2):
l=[i for i in xrange(len(lst)) if lst[i]=='something...']
И тогда (оба случая):
print(l)
Как и ожидалось.
Ответ 9
Если вы используете Python 2, вы можете добиться такой же функциональности:
f = lambda my_list, value:filter(lambda x: my_list[x] == value, range(len(my_list)))
Где my_list
- список, в котором вы хотите получить индексы, и value
- это искомое значение. Использование:
f(some_list, some_element)
Ответ 10
Если вам нужно искать все позиции элементов между определенными индексами, вы можете указать их:
[i for i,x in enumerate([1,2,3,2]) if x==2 & 2<= i <=3] # -> [3]
Ответ 11
Вы можете создать defaultdict
from collections import defaultdict
d1 = defaultdict(int) # defaults to 0 values for keys
unq = set(lst1) # lst1 = [1, 2, 2, 3, 4, 1, 2, 7]
for each in unq:
d1[each] = lst1.count(each)
else:
print(d1)
Ответ 12
Получение всех вхождений и положения одного или нескольких (идентичных) элементов в списке
С помощью enumerate (alist) вы можете сохранить первый элемент (n), который является индексом списка, когда элемент x равен тому, что вы ищете.
>>> alist = ['foo', 'spam', 'egg', 'foo']
>>> foo_indexes = [n for n,x in enumerate(alist) if x=='foo']
>>> foo_indexes
[0, 3]
>>>
Давайте сделаем нашу функцию findindex
Эта функция принимает элемент и список в качестве аргументов и возвращает позицию элемента в списке, как мы видели ранее.
def indexlist(item2find, list_or_string):
"Returns all indexes of an item in a list or a string"
return [n for n,item in enumerate(list_or_string) if item==item2find]
print(indexlist("1", "010101010"))
Выход
[1, 3, 5, 7]
просто
for n, i in enumerate([1, 2, 3, 4, 1]):
if i == 1:
print(n)
Выход:
0
4
Ответ 13
Использование filter() в python2.
>>> q = ['Yeehaw', 'Yeehaw', 'Googol', 'B9', 'Googol', 'NSM', 'B9', 'NSM', 'Dont Ask', 'Googol']
>>> filter(lambda i: q[i]=="Googol", range(len(q)))
[2, 4, 9]
Ответ 14
Изменить (идиотизм):
Добавление к хорошим ответам.
def count(x, lst):
ind = []
for i in lst:
if i == x:
ind.append(lst.index(x))
return ind