查找一个目录中所有以.txt为扩展名的文件的Python程序

2022-05-03 00:00:00 文件查找扩展名

在这个例子中，你将学习如何在一个目录中找到所有扩展名为.txt的文件。

例1：使用glob

import glob, os

os.chdir("my_dir")

for file in glob.glob("*.txt")
    print(file)

输出

c.txt
b.txt
a.txt

使用 glob 模块，你可以搜索具有特定扩展名的文件。

os.chdir("my_dir") 将当前工作目录设为 /my_dir。
使用for循环，你可以使用glob()搜索以.txt为扩展名的文件。 * 表示所有具有某个扩展名的文件。
例2：使用os

import os

for file in os.listdir("my_dir")
    if file.endwith(".txt"):
        print(file)

输出

a.txt
b.txt
c.txt

在这个例子中，我们使用endswith()方法来检查.txt的扩展名。

使用for循环，遍历/my_dir目录下的每个文件。
使用 endswith()方法检查该文件是否有扩展名.txt。
使用os.walk

import os

for root, dirs, files in os.walk("my_dir"):
    for file in files:
        if file.endwith(".txt")
            print(file)

输出

c.txt
b.txt
a.txt

这个例子使用了os模块的walk()方法。

使用for循环，遍历my_dir的每个文件。
使用 endswith()检查文件是否有扩展名.txt。

相关文章