如何使用glob2库选择特定的文本文件

时间:2018-12-19 01:21:17

标签: python

我一直在研究这段代码,我想知道如果可以选择特定文件而不是所有具有相同扩展名的文件

,该怎么办

import glob2
from datetime import datetime



filenames = glob2.glob("*.txt")
with open(datetime.now().strftime("%Y-%m-%d-%H-%M-%S-%f")+".txt", 'w') as file:
    for filename in filenames:
        with open(str(filename), "r") as f:
            file.write(f.read() + "\n")

我尝试将特定文件保存到元组或列表中,但不起作用

1 个答案:

答案 0 :(得分:1)

您可以尝试将文本文件存储在其中,并检查其中是否存在filenames中的文件:

import glob2

files = {'file1.txt', 'file2.txt'} # can use list or tuple here also

filenames = glob2.glob("*.txt")
for file in filenames:
    if file in files:
        # Do something with file

或者,您可以只使用os.listidr(),因为您只需要特定于 的文件:

from os import listdir

files = {'file1.txt', 'file2.txt'}

for file in listdir('.'):
    if file in files:
        # Do something with file