У меня есть список со значениями, встречающимися несколько раз. Я хочу перебрать список и проверить, встречается ли значение в первый раз.

Например: скажем, у меня есть один список, как,

L = ['a', 'a', 'a', 'b', 'b', 'b', 'b', 'b', 'e', 'e', 'e' .... ... ]

Теперь при каждом первом появлении элемента я хочу выполнить некоторый набор задач.

Как получить первое вхождение элемента?

Заранее спасибо!!

2
user3766871 28 Май 2019 в 16:23

2 ответа

Лучший ответ

Используйте set, чтобы проверить, обрабатывали ли вы уже этот элемент:

visited = set()
L = ['a','a','a','b','b','b','b','b','e','e','e'.......] 
for e in L:
    if e not in visited:
        visited.add(e)
        # process first time tasks
    else:
        # process not first time tasks
3
Netwave 28 Май 2019 в 13:32

Вы можете использовать unique_everseen из рецептов itertools.

Эта функция возвращает генератор, который выдает только первое вхождение элемента.

Код

from itertools import filterfalse

def unique_everseen(iterable, key=None):
    "List unique elements, preserving order. Remember all elements ever seen."
    # unique_everseen('AAAABBBCCDAABBB') --> A B C D
    # unique_everseen('ABBCcAD', str.lower) --> A B C D
    seen = set()
    seen_add = seen.add
    if key is None:
        for element in filterfalse(seen.__contains__, iterable):
            seen_add(element)
            yield element
    else:
        for element in iterable:
            k = key(element)
            if k not in seen:
                seen_add(k)
                yield element

Примере

lst = ['a', 'a', 'b', 'c', 'b']

for x in unique_everseen(lst):
    print(x) # Do something with the element

Выход

a
b
c

Функция unique_everseen также позволяет передавать key для сравнения элементов. Это полезно во многих случаях, например, если вам также необходимо знать положение каждого первого случая.

Примере

lst = ['a', 'a', 'b', 'c', 'b']

for i, x in unique_everseen(enumerate(lst), key=lambda x: x[1]):
    print(i, x)

Выход

0 a
2 b
3 c
0
Olivier Melançon 28 Май 2019 в 16:52