帮我写一段python代码,读取多个根目录路径三级文件夹中的文件,只读取txt格式文件,输出的文件的时候要输出一列来显示文件属于哪个一级文件夹,只显示出一级文件夹
时间: 2024-03-20 22:43:12 浏览: 50
python os用法总结
好的,以下是帮你实现的 Python 代码:
```python
import os
root_dirs = ['/path/to/root/dir1/', '/path/to/root/dir2/', '/path/to/root/dir3/']
for root_dir in root_dirs:
for dirpath, dirnames, filenames in os.walk(root_dir):
# 只遍历三级文件夹
if dirpath.count(os.sep) == 3:
# 获取一级文件夹名
first_dir = os.path.basename(os.path.dirname(os.path.dirname(dirpath)))
for filename in filenames:
# 只读取txt文件
if filename.endswith('.txt'):
file_path = os.path.join(dirpath, filename)
# 输出一列来显示文件属于哪个一级文件夹
print(first_dir, file_path)
```
你需要将代码中的 `/path/to/root/dir1/`、`/path/to/root/dir2/` 和 `/path/to/root/dir3/` 替换成你想要遍历的根目录路径列表,然后运行即可。代码会遍历所有三级文件夹,只读取其中的 txt 格式文件,并输出一列来显示文件属于哪个一级文件夹,只显示出一级文件夹。
阅读全文