Python中如何高效筛选第二元素重复的(Nx1)列表元素？

嘿，你的这个问题用双重循环确实效率很低——毕竟O(n²)的时间复杂度，数据量一大就会明显变慢。咱们换个思路，用哈希表（字典）分组的方式来解决，时间复杂度能降到O(n)，效率提升特别明显！

核心思路 #

我们可以把每个元组的第二个元素作为键，把对应的第一个元素收集到列表里。这样遍历一次列表就能完成分组，之后只要找出那些列表长度大于1的键，对应的所有第一个元素就是你要的结果。

具体实现（Python） #

方法1：用collections.defaultdict简化分组

arr = [(2,3),(4,3),(3,4),(1,4),(5,4),(6,5)]
from collections import defaultdict

# 初始化一个默认值为列表的字典
grouped = defaultdict(list)
for first_val, second_val in arr:
    grouped[second_val].append(first_val)

# 筛选并输出符合条件的元素
for key, values in grouped.items():
    if len(values) > 1:
        print("重复键{}对应的第一个元素：".format(key), values)

运行后会输出：

重复键3对应的第一个元素： [2, 4]
重复键4对应的第一个元素： [3, 1, 5]

方法2：不用额外库，用普通字典实现

如果你不想导入collections，也可以手动处理字典的初始化：

arr = [(2,3),(4,3),(3,4),(1,4),(5,4),(6,5)]
grouped = {}

for first_val, second_val in arr:
    if second_val not in grouped:
        grouped[second_val] = []
    grouped[second_val].append(first_val)

# 同样的筛选逻辑
for key, values in grouped.items():
    if len(values) > 1:
        print(', '.join(map(str, values)))

如果你需要收集结果而不是直接打印

可以用列表推导式把所有符合条件的元素整理到一个列表里：

result = [val for key, vals in grouped.items() if len(vals) > 1 for val in vals]
print(result)  # 输出: [2, 4, 3, 1, 5]

为什么这个方法更快？ #

原来的双重循环需要两两比较每个元素，数据量越大，重复比较的次数会指数级增长。而字典分组的方式只需要遍历两次列表（一次分组，一次筛选），每个元素只被处理一次，不管列表有多长，效率都能保持稳定。

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