用python编写一个程序,要求用户输入一个文件夹路径和一个文件扩展名,程序需要找出该文件夹中所有该扩展名的文件,并将它们复制到新的文件夹中。
时间: 2024-05-01 08:20:08 浏览: 89
可以使用Python的os和shutil模块来实现该任务。下面是一个示例代码:
```python
import os
import shutil
# 获取用户输入的文件夹路径和文件扩展名
folder_path = input("请输入文件夹路径:")
file_ext = input("请输入文件扩展名(例如:.txt):")
# 获取文件夹中所有该扩展名的文件
files = [f for f in os.listdir(folder_path) if f.endswith(file_ext)]
# 创建新文件夹
new_folder_path = os.path.join(folder_path, "new_folder")
os.makedirs(new_folder_path, exist_ok=True)
# 复制文件到新文件夹中
for f in files:
src_path = os.path.join(folder_path, f)
dst_path = os.path.join(new_folder_path, f)
shutil.copy(src_path, dst_path)
print("文件复制完成!")
```
该程序首先获取用户输入的文件夹路径和文件扩展名,然后使用os模块的listdir函数获取文件夹中所有文件的文件名,再使用列表推导式筛选出所有该扩展名的文件。接着使用os模块的join函数创建新文件夹,并使用shutil模块的copy函数将文件复制到新文件夹中。最后输出完成提示。注意,如果新文件夹已经存在,则使用exist_ok=True参数可以避免抛出FileExistsError异常。
阅读全文