编写函数,获取指定文件夹下的.py文件列表
时间: 2024-02-28 17:55:31 浏览: 72
python实现读取文件夹下文件列表
好的,可以使用Python自带的os模块来实现。以下是获取指定文件夹下所有.py文件的函数:
```python
import os
def get_python_files(path):
"""
获取指定文件夹下的.py文件列表
:param path: 文件夹路径
:return: .py文件列表
"""
py_files = []
for root, dirs, files in os.walk(path):
for file in files:
if file.endswith('.py'):
py_files.append(os.path.join(root, file))
return py_files
```
使用时,只需传入要查询的文件夹路径即可:
```python
path = '/path/to/folder'
python_files = get_python_files(path)
print(python_files)
```
注意修改`path`变量的值为你要查询的文件夹路径。
阅读全文