判断子目录并列出子目录中的python文件
时间: 2024-03-21 22:39:03 浏览: 19
可以使用Python中的os模块和glob模块来实现。具体的代码如下:
```python
import os
import glob
root_dir = "C:/Users/username/Documents"
# 列出根目录下的所有子目录
sub_dirs = [os.path.join(root_dir, d) for d in os.listdir(root_dir) if os.path.isdir(os.path.join(root_dir, d))]
# 遍历每个子目录并列出其中的.py文件
for sub_dir in sub_dirs:
print("子目录:", sub_dir)
py_files = glob.glob(os.path.join(sub_dir, "*.py"))
for py_file in py_files:
print(" Python文件:", py_file)
```
其中,我们首先通过`os.listdir(root_dir)`函数列出根目录下的所有子目录的名称;然后使用列表推导式将每个子目录的完整路径存储到`sub_dirs`列表中。接着,我们遍历每个子目录,使用`glob.glob(os.path.join(sub_dir, "*.py"))`函数列出该子目录中所有的.py文件,存储到`py_files`列表中。最后,我们将子目录的完整路径和其中的Python文件逐一输出。
需要注意的是,在使用`os.path.join(root_dir, d)`函数连接路径时,需要使用`root_dir`的绝对路径,否则可能会出现路径连接错误的情况。