Как удалить элементы списка в зависимости от предшественника в python
Учитывая список Python, я хочу удалить последовательные «дубликаты». Однако повторяющееся значение является атрибутом элемента списка (в этом примере это первый элемент tuple).
Вход:
[(1, 'a'), (2, 'b'), (2, 'b'), (2, 'c'), (3, 'd'), (2, 'e')]
Желаемый результат:
[(1, 'a'), (2, 'b'), (3, 'd'), (2, 'e')]
Нельзя использовать set или dict, потому что важен порядок.
Невозможно использовать понимание списка [x for x in somelist if not determine(x)], потому что проверка зависит от предшественника.
Я хочу что-то вроде:
mylist = [...]
for i in range(len(mylist)):
if mylist[i-1].attr == mylist[i].attr:
mylist.remove(i)
Каков предпочтительный способ решения этой проблемы в Python?
Возможный дубликат Словари: как сохранить ключи/значения в том же порядке, что и объявленный?
Вас интересуют только дубликаты последовательный? То есть, если бы последний элемент списка также был 1, 'a', был бы он дубликатом первого?
Какой должен быть результат для [(1, 'a'), (2, 'a'), (1, 'a')] ? Должно быть [(1, 'a'), (2, 'a'), (1, 'a')] или [(1, 'a'), (2, 'a')]?
Да, речь идет о последовательных дубликатах. Вывод для [(1, 'a'), (2, 'a'), (1, 'a')] должен быть [(1, 'a'), (2, 'a'), (1, 'a')].
Считаются ли (1, 'a') и (1, 'b') равными?
@gmds, да, (1, 'a') и (1, 'b') в этом случае считаются одинаковыми дубликатами.
@Sparkofska В этом случае вам нужно будет использовать мое решение, в котором указана ключевая функция для сравнения только с первым элементом tuple.
@gmds Ты прав. Принятый и отредактированный пример соответственно.
Ответы 8
Вы можете использовать itertools.groupby (демонстрация с дополнительными данными):
from itertools import groupby
from operator import itemgetter
data = [(1, 'a'), (2, 'a'), (2, 'b'), (3, 'a'), (4, 'a'), (2, 'a'), (2, 'a'), (3, 'a'), (3, 'a')]
[next(group) for key, group in groupby(data, key=itemgetter(0))]
Выход:
[(1, 'a'), (2, 'a'), (3, 'a'), (4, 'a'), (2, 'a'), (3, 'a')]
Для полноты итеративный подход, основанный на других ответах:
result = []
for first, second in zip(data, data[1:]):
if first[0] != second[0]:
result.append(first)
result
Выход:
[(1, 'a'), (2, 'b'), (3, 'a'), (4, 'a'), (2, 'a')]
Обратите внимание, что это сохраняет дубликат прошлой вместо первого.
Вам не нужен ключевой параметр, просто возьмите ключ каждой группы
@yatu В вопросе говорится, что «дубликатное значение является атрибутом list», что означает, что это не сработает, если (2, 'a') и (2, 'b') считаются равными.
Я вижу, да, в этом случае действительно имеет смысл @gmds. Однако на этом примере трудно сказать. ИМО, если бы это было то, что имел в виду ОП, более общий пример имел бы больше смысла.
Если я не ошибаюсь, вам нужно искать только последнее значение.
test = [(1, 'a'), (2, 'a'), (2, 'a'), (3, 'a'), (4, 'a'),(3, 'a'),(4,"a"),(4,"a")]
result = []
for i in test:
if result and i[0] == result[-1][0]: #edited since OP considers (1,"a") and (1,"b") as duplicate
#if result and i == result[-1]:
continue
else:
result.append(i)
print (result)
Выход:
[(1, 'a'), (2, 'a'), (3, 'a'), (4, 'a'), (3, 'a'), (4, 'a')]
Я бы немного изменил предложение Генри Йика, сделав его немного проще. Не уверен, что я что-то упускаю.
inputList = [(1, 'a'), (2, 'a'), (2, 'a'), (3, 'a'), (2, 'a')]
outputList = []
lastItem = None
for item in inputList:
if not item == lastItem:
outputList.append(item)
lastItem = item
print(outputList)
Вы можете легко zip список с собой. Каждый элемент, кроме первого, заархивирован со своим предшественником:
>>> L = [(1, 'a'), (2, 'b'), (2, 'b'), (2, 'c'), (3, 'd'), (2, 'e')]
>>> list(zip(L[1:], L))
[((2, 'b'), (1, 'a')), ((2, 'b'), (2, 'b')), ((2, 'c'), (2, 'b')), ((3, 'd'), (2, 'c')), ((2, 'e'), (3, 'd'))]
Первый элемент всегда является частью результата, а затем вы фильтруете пары по условию и возвращаете первый элемент:
>>> [L[0]]+[e for e, f in zip(L[1:], L) if e[0]!=f[0]]
[(1, 'a'), (2, 'b'), (3, 'd'), (2, 'e')]
Если вы просто хотите придерживаться понимания списка, вы можете использовать что-то вроде этого:
>>> li = [(1, 'a'), (2, 'a'), (2, 'a'), (3, 'a'), (2, 'a')]
>>> [li[i] for i in range(len(li)) if not i or li[i] != li[i-1]]
[(1, 'a'), (2, 'a'), (3, 'a'), (2, 'a')]
Пожалуйста, обратите внимание, что not i — это питонический способ написания i == 0.
Вы также можете использовать enumerate и понимание списка:
>>> data = [(1, 'a'), (2, 'b'), (2, 'b'), (2, 'c'), (3, 'd'), (2, 'e')]
>>> [v for ix, v in enumerate(data) if not ix or v[0] != data[ix-1][0]]
[(1, 'a'), (2, 'b'), (3, 'd'), (2, 'e')]
Хороший, потому что не нужно никаких imports. Также v[0] можно заменить любым v.get_attribute(), что делает его достаточно универсальным.
Это несколько излишне, но вы также можете использовать «уменьшить»:
from functools import reduce
data=[(1, 'a'), (2, 'b'), (2, 'b'), (2, 'c'), (3, 'd'), (2, 'e')]
reduce(lambda rslt,t: rslt if rslt[-1][0]==t[0] else rslt+[t], data, [data[0]])
[(1, 'a'), (2, 'b'), (3, 'd'), (2, 'e')]