编写函数,获取指定文件夹下的.py文件列表
时间: 2024-02-28 21:55:34 浏览: 50
python 获取指定文件夹下所有文件名称并写入列表的实例
5星 · 资源好评率100%
好的,您可以使用Python内置的os和glob模块来实现这个功能。下面是一个获取指定文件夹下的.py文件列表的函数:
```python
import os
import glob
def get_py_files(folder_path):
os.chdir(folder_path)
py_files = glob.glob("*.py")
return py_files
```
这个函数需要传入一个文件夹路径作为参数,然后会使用os模块切换到指定路径。接着,使用glob模块查找所有以.py结尾的文件,并将它们存储在一个列表中。最后,返回这个列表。
您可以使用以下代码调用这个函数:
```python
folder_path = "/path/to/folder"
py_files = get_py_files(folder_path)
print(py_files)
```
请将上面代码中的“/path/to/folder”替换为您想要查找.py文件的文件夹路径。这个函数将返回一个包含所有.py文件名的列表,您可以根据需要进一步处理这个列表。
阅读全文