Как удалить элементы списка в зависимости от предшественника в 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
Вы можете использовать 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')]
Обратите внимание, что это сохраняет последний дубликат, а не первый.
Ответ 2
Чтобы удалить последовательные дубликаты, вы можете использовать itertools.groupby
:
l = [(1, 'a'), (2, 'a'), (2, 'a'), (3, 'a'), (4, 'a')]
from itertools import groupby
[tuple(k) for k, _ in groupby(l)]
# [(1, 'a'), (2, 'a'), (3, 'a'), (4, 'a')]
Ответ 3
Если я не ошибаюсь, вам нужно только найти последнее значение.
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')]
Ответ 4
Если вы просто хотите придерживаться понимания списка, вы можете использовать что-то вроде этого:
>>> 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
.
Ответ 5
Вы также можете использовать 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')]
Ответ 6
Я бы немного изменил предложение Генри Йика, сделав его немного проще. Не уверен, что я что-то упустил.
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)
Ответ 7
Вы можете легко 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')]
Ответ 8
Это несколько излишне, но вы также можете использовать "уменьшить":
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')]