python - 如果元素以另一个列表中的元素结尾，则从列表中删除元素的结尾

Question

我有以下两个列表。如果 my_list 结束 带有扩展名的扩展名，则应将其删除。我似乎找不到不需要太多代码行的解决方案。

输入:

my_list = ['abc_sum_def_sum', 'abc_sum_def_mean', 'abc_sum', 'abc_abc']

extensions = ['_sum', '_mean']

输出:

new_list = ['abc_sum_def', 'abc_sum_def', 'abc', 'abc_abc']

Answer 1

为了获得更好的pythonic方法，您可以将其转换为列表理解:

my_list = ['abc_sum_def_sum','abc_sum_def_mean','abc_sum','abc_abc']

extensions = ['_sum','_mean']
new_list =[]
for x in my_list:
    for elem in extensions:
        if x.endswith(elem):
            y = x[:-len(elem)]
            new_list.append(y)