我有一个 python 脚本，其中列出了具有特定扩展名的文件。 我想将这些文件的文件名打印到 txt 文件中。 都在同一条路上

Question

我有这个路径C:\Users\PacY\Documents它包含文件和文件夹。 我有一个 python 脚本，它可以打印出扩展名为pdf 、 docx 、 jpg的文件。 我想将具有上述扩展名的那些文件的文件名添加到位于同一路径C:\Users\PacY\Documents的名为“extension_files”的 txt 文件中。

代码如下。

import os
path = r'C:\Users\PacY\Documents'

extensions = ['.pdf', '.docx', '.jpg']


for file in os.listdir(path):
    for extension in extensions:
        if file.__contains__(extension):
            print(file)

Answer 1

试试这个

import os

path = r'C:\Users\PacY\Documents'

extensions = ['.pdf', '.docx', '.jpg']

txt_file = r"C:\Users\PacY\Documents\found_files\txt.txt"  # name of your txt file

with open(txt_file, "a+") as txt:
    for file in os.listdir(path):
        for extension in extensions:
            if file.__contains__(extension):
                txt.write(file)
                txt.write("\n")