如何過濾帶有特定前綴和後綴（擴展名）的文件名？

Question

我有這樣的文件列表：

file_list = ['file1.zip', 'file1.txt'] 
file_prefix = 'file1' 

我想使用filter和re只得到file1.txt以上。我試試這個：

regex = re.compile(file_prefix + '.*(!zip).*') 
result = list(filter(regex.search, file_list)) 
# in the above, result should be populated with just ['file1.txt'] 

但是，正則表達式模式不起作用。有人可以幫我解決這個問題嗎？非常感謝先進！

Answer 1

可以使用負前瞻是這樣的：

regex = re.compile(file_prefix + '(?!\.zip)') 

代碼：

>>> file_list = ['file1.zip', 'file1.txt'] 
>>> file_prefix = 'file1' 
>>> regex = re.compile(file_prefix + '(?!\.zip)') 
>>> print list(filter(regex.search, file_list)) 
['file1.txt'] 

(?!\.zip)使得當.zip根本不存在下一個位置斷言真正的負先行。

Read more about look-arounds

Answer 2

無需正則表達式這個解決方案 - 你不需要把一門大炮來拇指鬥爭。使用Python的本地字符串搜索/檢查：

file_list = ["file1.zip", "file1.txt"] 
file_prefix = "file1" 
file_exclude = "zip" 

result = [e for e in file_list if e.startswith(file_prefix) and not e.endswith(file_exclude)] 
# ['file1.txt'] 

也應該快得多。

如果不想僅搜索邊緣，要過濾出沒有zip後綴file_prefix不管它是字符串中（所以你要匹配some_file1.txt後的條目，甚至a_zip_file1.txt，但不是​​），你可以稍微修改：

file_list = ["file1.zip", "file1.txt", "some_file1.txt", "a_zip_file1.txt", "file1_zip.txt"] 
file_prefix = "file1" 
file_exclude = "zip" 

result = [e for e in file_list if e.find(file_exclude) < e.find(file_prefix)] 
# ['file1.txt', 'some_file1.txt', 'a_zip_file1.txt']