如何基于元组索引0去重列表元组并保留首次出现及原顺序

你现在遇到的问题，核心是想用itertools.groupby处理但没踩对它的特性，而且random.choice也不符合你“保留首次出现”的需求——我来给你拆解下问题，再给你合适的解法：

首先说你现有代码的问题：

• groupby只会把连续拥有相同key的元素归为一组，你的原列表里索引0为1的元组不是连续的，所以不会被分到同一组，自然没法去重；
• random.choice是随机选组内元素，而你要的是保留每个索引0值第一次出现的元组，完全不需要随机。

推荐解法（保持原列表顺序） #

用一个集合来跟踪已经出现过的索引0值，遍历原列表时只保留第一次遇到的元素，这样既能去重，又完全维持原顺序：

abc = [(1,2,3), (2,3,4), (1,0,3),(0,2,0), (2,4,5),(5,4,3), (0,4,1)]
seen_keys = set()
new_list = []

for item in abc:
    current_key = item[0]
    if current_key not in seen_keys:
        seen_keys.add(current_key)
        new_list.append(item)

print(new_list)

运行后会得到你期望的输出：[(1, 2, 3), (2, 3, 4), (0, 2, 0), (5, 4, 3)]

如果你坚持想用groupby（会打乱原顺序） #

如果不在意原列表的顺序，也可以先按索引0排序，再用groupby分组后取每组第一个元素：

from itertools import groupby

abc = [(1,2,3), (2,3,4), (1,0,3),(0,2,0), (2,4,5),(5,4,3), (0,4,1)]
# 先按索引0排序，让同key的元素连续
sorted_abc = sorted(abc, key=lambda x: x[0])
new_list = [next(group) for _, group in groupby(sorted_abc, key=lambda x: x[0])]

print(new_list)

不过这个方法会改变原列表的元素顺序，所以更推荐第一种解法。

内容的提问来源于stack exchange，提问作者A.S