python - 替换列表中的字符串(使用 re.sub)

Question

我正在尝试替换文件列表中的部分文件扩展名。我希望能够遍历项目(文件)，并删除扩展名。当 re.sub 作为第三个参数需要一个字符串时，我不知道如何适本地循环遍历列表中的项目。例如。 re.sub(模式，repl，字符串，计数= 0，标志= 0)

import re

file_lst = ['cats1.fa', 'cats2.fa', 'dog1.fa', 'dog2.fa']
file_lst_trimmed =[]

for file in file_lst:
    file_lst_trimmed = re.sub(r'1.fa', '', file)

这里出现的问题是 re.sub 需要一个字符串，我希望它遍历字符串列表。

感谢您的任何建议!

Answer 1

您可以使用列表理解来构造具有清理过的文件名的新列表。 \d是匹配单个字符和 $ 的正则表达式只匹配字符串的末尾。

file_lst_trimmed = [re.sub(r'\d\.fa$', '', file) for file in file_lst]

结果:

>>> file_lst_trimmed 
['cats', 'cats', 'dog', 'dog']