如何简洁实现：移除列表元素末尾匹配另一列表的后缀

这有个非常紧凑的一行式解法，完美匹配你的需求，不需要额外的复杂逻辑：

my_list = ['abc_sum_def_sum', 'abc_sum_def_mean', 'abc_sum', 'abc_abc']
extensions = ['_sum', '_mean']

new_list = [next((s[:-len(ext)] for ext in extensions if s.endswith(ext)), s) for s in my_list]

运行后new_list就是你要的结果：['abc_sum_def', 'abc_sum_def', 'abc', 'abc_abc']

为什么这个方法好用？ #

• 用列表推导式+生成器的组合，代码极简，可读性也不错
• next(..., s)是关键：遍历extensions找到第一个匹配的后缀，就截断字符串；如果没有任何后缀匹配，直接返回原字符串
• 完全避免了冗余的循环和条件判断，一行搞定所有处理

如果偏好正则表达式的方式，也可以用这个简洁的版本（需要导入re模块）：

import re

pattern = re.compile(r'(' + '|'.join(re.escape(ext) for ext in extensions) + r')$')
new_list = [pattern.sub('', s) for s in my_list]

这个方法通过构造匹配结尾后缀的正则，一次性替换掉匹配的部分，同样很简洁。

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