index()목록에서 항목의 첫 항목 만 제공합니다. 목록의 모든 인덱스를 반환하는 깔끔한 트릭이 있습니까?

목록 이해를 사용할 수 있습니다.

indices = [i for i, x in enumerate(my_list) if x == "whatever"]

직접 목록에 대한 해결책은 아니지만 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])

이것은 다른 솔루션에 비해 많은 수의 요소가있는 목록 (배열)의 경우 훨씬 빠릅니다.

다음을 사용하는 솔루션 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 개의 정수 목록이있는 테스트에서) 목록 이해가 더 빠릅니다.

어때요 :

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]

occurrences = lambda s, lst: (i for i,e in enumerate(lst) if e == s)
list(occurrences(1, [1,2,3,1])) # = [0, 3]

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입니다.

모든 경우에 대한 하나 이상의 솔루션 (중복 된 경우 죄송합니다) :

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)

또는 사용 range(파이썬 3) :

l=[i for i in range(len(lst)) if lst[i]=='something...']

(파이썬 2)의 경우 :

l=[i for i in xrange(len(lst)) if lst[i]=='something...']

그리고 (두 경우 모두) :

print(l)

예상대로입니다.

python2에서 filter () 사용.

>>> q = ['Yeehaw', 'Yeehaw', 'Googol', 'B9', 'Googol', 'NSM', 'B9', 'NSM', 'Dont Ask', 'Googol']
>>> filter(lambda i: q[i]=="Googol", range(len(q)))
[2, 4, 9]

당신은 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)

모든 발생 및 목록에서 하나 이상의 (동일한) 항목 위치 가져 오기

enumerate (alist)를 사용하면 요소 x가 찾는 것과 같을 때 목록의 인덱스 인 첫 번째 요소 (n)를 저장할 수 있습니다.

>>> 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

사용하여 for-loop:

• 답변 enumerate및 목록 이해 보다 효율적이고 파이썬, 그러나,이 답변은 그 중 일부를 사용 할 수 없습니다 학생들을 대상으로 내장 된 기능을 .
• 빈 목록을 만들고 indices
• 와 루프를 만들 for i in range(len(x)): 기본적으로 인덱스 위치 목록을 반복합니다.[0, 1, 2, 3, ..., len(x)-1]
• 루프에서에 일치하는 i곳 을x[i]valueindices
  • x[i] 색인으로 목록에 액세스

def get_indices(x: list, value: int) -> list:
    indices = list()
    for i in range(len(x)):
        if x[i] == value:
            indices.append(i)
    return indices

n = [1, 2, 3, -50, -60, 0, 6, 9, -60, -60]
print(get_indices(n, -60))

>>> [4, 8, 9]

• 함수 get_indices는 type hints 로 구현됩니다 . 이 경우, 목록에서 n의 무리입니다 int들, 그러므로 우리는 검색 value도 정의,int .

사용 while-loop및 .index:

• 로 .index사용 try-except에 대한 오류 처리 a는 때문에 ValueError경우에 발생합니다 value목록에 없습니다.

def get_indices(x: list, value: int) -> list:
    indices = list()
    i = 0
    while True:
        try:
            # find an occurrence of value and update i to that index
            i = x.index(value, i)
            # add i to the list
            indices.append(i)
            # advance i by 1
            i += 1
        except ValueError as e:
            break
    return indices

print(get_indices(n, -60))
>>> [4, 8, 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)

특정 색인 사이의 모든 요소 위치를 검색해야하는 경우 다음을 지정할 수 있습니다.

[i for i,x in enumerate([1,2,3,2]) if x==2 & 2<= i <=3] # -> [3]