如何将Python元组匹配的嵌套for循环改写为更Pythonic的代码？

核心思路 #

将嵌套循环改写为集合推导式直接生成目标集合，既简洁贴合Pythonic风格，又能避免手动调用集合add方法的繁琐；如果处理数据量较大，还可以通过字典索引优化查找效率。

优化实现方案 #

方案1：集合推导式（简洁直观） #

直接用集合推导式替代嵌套循环，一步生成结果集合：

set1 = {(1, 'a'), (2, 'b'), (3, 'c')}
set2 = {(1, 'x'), (2, 'y'), (4, 'z')}

result = {(item1[1], item2[1]) for item1 in set1 for item2 in set2 if item1[0] == item2[0]}
print(result)  # 输出：{('a', 'x'), ('b', 'y')}

方案2：字典索引优化（高效处理大数据） #

当集合元素数量较多时，嵌套循环的时间复杂度为O(n*m)，可将其中一个集合转为字典（以元组第一个元素为键），将查找复杂度降至O(n+m)：

set1 = {(1, 'a'), (2, 'b'), (3, 'c')}
set2 = {(1, 'x'), (2, 'y'), (4, 'z')}

# 转换set1为字典，键是元组第一个元素，值是第二个元素
dict_from_set1 = {key: val for key, val in set1}
# 遍历set2，匹配存在的键并生成结果集合
result = {(dict_from_set1[key], val) for key, val in set2 if key in dict_from_set1}
print(result)  # 输出：{('a', 'x'), ('b', 'y')}

关于列表推导式的问题解决 #

你之前用列表推导式得到列表后无法调用集合add方法，只需将列表推导式直接转为集合即可：

# 列表推导式转集合的写法
result = set([(item1[1], item2[1]) for item1 in set1 for item2 in set2 if item1[0] == item2[0]])

不过这种写法会先创建列表再转集合，效率略低于直接使用集合推导式，更推荐前者。

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