Python中基于末尾3位共享子串筛选合并列表的高效方法

需求说明 #

给定两个列表：

list1 = ["Time1Person001", "Time1Person002", "Time1Person003", "Time1Person004", "Time1Person005"]
list2 = ["Time2Person001", "Time2Person003", "Time2Person004", "Time2Person007"]

需要生成列表list3，只保留两个列表中末尾3位字符串相同的元素——即某个元素要被保留，必须存在另一个列表里的元素和它末尾3位一致。预期输出：

list3 = ["Time1Person001", "Time1Person003", "Time1Person004", "Time2Person001", "Time2Person003", "Time2Person004"]

高效实现方法 #

要避免低效的双重循环比对，最优方案是利用集合的快速查询特性，把时间复杂度降到O(n+m)（n、m分别是两个列表的长度）。

代码实现 #

方法一：先取两个列表后缀的交集，再筛选

这个方法逻辑最严谨，完全匹配“两个列表中都存在对应后缀”的需求：

# 提取两个列表所有元素的末尾3位，存入集合
suffix_set1 = {item[-3:] for item in list1}
suffix_set2 = {item[-3:] for item in list2}

# 找出两个集合的交集——也就是同时出现在两个列表里的后缀
common_suffixes = suffix_set1 & suffix_set2

# 遍历两个列表的所有元素，筛选出末尾3位在交集中的项
list3 = [item for item in list1 + list2 if item[-3:] in common_suffixes]

print(list3)

方法二：基于其中一个列表的后缀集合筛选（代码更简洁）

如果需求明确是“保留list1全部元素，加上list2中与list1后缀匹配的元素”，可以用这个更短的写法：

# 用list1的后缀创建查询集合
valid_suffixes = {s[-3:] for s in list1}

# 直接合并list1和list2中符合条件的元素
list3 = list1 + [item for item in list2 if item[-3:] in valid_suffixes]

print(list3)

方案优势 #

• 集合的成员查询是**平均O(1)**的时间复杂度，远快于列表的O(n)查询；
• 整个流程仅需遍历两个列表各一次，加上集合的交集/创建操作（均为线性时间），整体时间复杂度为O(n+m)，相比双重循环的O(n*m)，在列表元素较多时效率提升极其明显。

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