将符合条件的文件路径列表保存到一个文本文件中,并返回文件路径
时间: 2024-05-16 07:13:31 浏览: 16
以下是 Python 代码示例,可以实现将符合条件的文件路径列表保存到一个文本文件中,并返回文件路径:
```python
import os
def save_file_paths(root_folder, file_extension, output_file):
file_paths = []
for root, dirs, files in os.walk(root_folder):
for file in files:
if file.endswith(file_extension):
file_path = os.path.join(root, file)
file_paths.append(file_path)
with open(output_file, 'w') as f:
for file_path in file_paths:
f.write(file_path + '\n')
return os.path.abspath(output_file)
```
这个函数 `save_file_paths` 接受 3 个参数:
- `root_folder`:要搜索的文件夹的路径。
- `file_extension`:要搜索的文件扩展名。
- `output_file`:保存文件路径列表的文本文件路径。
函数使用 `os.walk` 函数遍历目录树,并使用 `endswith` 函数检查文件名是否以指定的扩展名结尾。如果文件名符合条件,函数就将文件路径添加到 `file_paths` 列表中。
然后,函数使用 `open` 函数创建一个新文件,并使用 `write` 函数将文件路径列表写入文件中。最后,函数返回输出文件的绝对路径。
你可以像这样调用函数:
```python
root_folder = '/path/to/folder'
file_extension = '.txt'
output_file = '/path/to/output_file.txt'
output_file_path = save_file_paths(root_folder, file_extension, output_file)
print('保存文件路径列表到文件:', output_file_path)
```
这会在指定的文件夹中搜索所有扩展名为 `.txt` 的文件,并将文件路径列表保存到 `/path/to/output_file.txt` 文件中。函数会返回输出文件的绝对路径。