您可以使用Python的os模块中的函数来遍历目录并获取子文件夹和特定文件。下面是代码示例,其中我们将搜索名为“test”的文件夹并在其中查找扩展名为“.txt”的文件。
import os
# 获取当前工作目录
dir_path = os.getcwd()
# 遍历文件夹以及子文件夹中所有文件和文件夹
for root, dirs, files in os.walk(dir_path):
# 迭代子文件夹
for dir in dirs:
print(os.path.join(root, dir))
# 迭代特定扩展名的文件
for file in files:
if file.endswith(".txt"):
print(os.path.join(root, file))
在上面的示例中,os.walk()函数返回一个迭代器,该迭代器在遍历目录时生成根(root)、子文件夹(dirs)和文件(files)。我们使用os.path.join()函数来获取完整路径,并使用.endswith()方法来检查文件扩展名是否匹配。
运行此代码,它将打印出所述目录及其子文件夹的名称,以及所有扩展名为“.txt”的文件的名称。