怎么设置os.walk函数只遍历一次呢
时间: 2023-05-30 07:03:32 浏览: 268
Python使用os.listdir()和os.walk()获取文件路径与文件下所有目录的方法
5星 · 资源好评率100%
os.walk函数本身只会遍历一次,即遍历整个目录树并返回所有文件和子目录的列表。如果你想控制os.walk函数只遍历某些特定的文件或目录,可以使用if语句来过滤掉不需要的文件或目录。
例如,以下代码只遍历根目录下的所有目录和.py文件:
```python
import os
root_dir = '/path/to/root/dir'
for dirpath, dirnames, filenames in os.walk(root_dir):
if not dirpath.startswith(root_dir):
continue
# 遍历所有.py文件
for filename in filenames:
if filename.endswith('.py'):
print(os.path.join(dirpath, filename))
```
注意,以上代码仅供参考,具体的文件和目录过滤规则根据实际需求而定。
阅读全文