解决这个问题的一种方法是使用递归来遍历目录,并找到所有数据文件。以下是一个示例代码:
import os
def find_data_files(directory):
data_files = []
for root, dirs, files in os.walk(directory):
for file in files:
if file.endswith('.csv') or file.endswith('.txt'):
data_files.append(os.path.join(root, file))
return data_files
# 指定目录
directory = '/path/to/data/files'
# 查找数据文件
data_files = find_data_files(directory)
# 打印找到的文件列表
for file in data_files:
print(file)
这段代码使用os.walk
函数遍历目录树,并在每个目录中查找以.csv
或.txt
结尾的文件。找到的文件路径将被添加到data_files
列表中。最后,可以遍历data_files
列表并打印出找到的文件路径。
请注意,上述代码假设目录中只包含数据文件,如果目录中还包含其他类型的文件,可以根据需要进行适当的修改。