Python：如何找出两个元组列表前n个元素中首元素相同的项（n=3）

这里有个简洁的Python实现思路，一步步来给你拆解：

步骤1：先明确输入示例（方便演示） #

假设我们有两个元组列表作为输入：

list1 = [('AB', 1), ('CD', 2), ('EF', 3), ('GH', 4)]
list2 = [('AB', 5), ('EF', 6), ('IJ', 7), ('KL', 8)]
n = 3

步骤2：截取每个列表的前n个元素 #

用Python的切片语法[:n]就能轻松获取前3个元素——哪怕列表长度不足3，切片也会自动取到所有现有元素，不会报错：

# 处理两个列表，只保留前n个元素
processed_list1 = list1[:n]
processed_list2 = list2[:n]

步骤3：提取首元素并找交集 #

我们可以用集合来快速找出共同的首元素，集合的交集操作效率很高，非常适合这种场景：

# 分别提取两个处理后列表中所有元组的首元素，转成集合
keys1 = {t[0] for t in processed_list1}
keys2 = {t[0] for t in processed_list2}

# 取两个集合的交集，再转成列表就是结果
result = list(keys1 & keys2)

运行后result会得到['AB', 'EF']（注意集合是无序的，如果需要保持元素在原列表中的出现顺序，可以看下面的优化方案）

优化：保持结果的顺序（按原列表出现顺序） #

如果希望结果的顺序和list1里前n个元素的首元素出现顺序一致，我们可以遍历processed_list1的首元素，检查是否在另一个集合里：

result = [t[0] for t in processed_list1 if t[0] in keys2]

这样得到的结果会严格按照list1里前3个元组首元素的顺序排列，比如如果list1前3个是('EF',3), ('AB',1), ('CD',2)，结果就会是['EF','AB']

边界情况处理 #

• 如果其中一个列表的元素数量少于n，切片[:n]会自动取全部元素，不影响后续逻辑
• 如果没有任何相同的首元素，结果会返回空列表[]

内容的提问来源于stack exchange，提问作者Dawn17