用python语言编写一个程序,要求用户输入一个文件夹路径和一个文件扩展名,程序需要找出该文件夹中所有该扩展名的文件,并将它们复制到新的文件夹中。写出注释和思路
时间: 2023-06-24 12:08:05 浏览: 92
首先,我们需要引入 os 和 shutil 两个 Python 模块来实现文件操作和复制操作。然后,我们需要用 input() 函数来获取用户输入的文件夹路径和文件扩展名。接着,我们可以使用 os.listdir() 函数来列出文件夹中的所有文件和文件夹,使用 os.path.splitext() 函数来获取文件的扩展名,并使用 shutil.copy() 函数将符合条件的文件复制到新的文件夹中。
以下是代码实现及注释:
```python
import os
import shutil
# 获取用户输入的文件夹路径和文件扩展名
folder_path = input("请输入文件夹路径:")
file_extension = input("请输入文件扩展名(如 .txt):")
# 新建一个文件夹用于存放符合条件的文件
new_folder_path = os.path.join(folder_path, "new_folder")
if not os.path.exists(new_folder_path):
os.mkdir(new_folder_path)
# 遍历文件夹中的所有文件和文件夹
for filename in os.listdir(folder_path):
# 获取文件的扩展名
_, extension = os.path.splitext(filename)
# 判断文件是否符合条件
if extension == file_extension:
# 构造源文件路径和目标文件路径
src_file_path = os.path.join(folder_path, filename)
dst_file_path = os.path.join(new_folder_path, filename)
# 复制文件到新的文件夹中
shutil.copy(src_file_path, dst_file_path)
print("复制完成!")
```
阅读全文