《Python编程快速上手》选择性拷贝

问题如下：

编写一个程序，遍历一个目录树，查找特的那个拓展名的文件（如，.jpg或.pdf）。不论这些文件位置在哪里，将它们拷贝到一个新的文件夹中

思路如下：

- 程序需要做以下事情：
遍历文件目录树，并返回相关文件夹及文件信息
查找特定文件名文件
将找到的特定文件拷贝到新文件夹
- 代码需要做以下事情：
导入os,shutil模块
input()输入需要查找的文件拓展名,遍历的文件夹及复制目标文件夹
os.walk(0遍历目录树
filename.endswith(’’)查找特定文件名文件
shutil模块移动，复制相应文件
拷贝完成反馈结果

---

代码如下：

import os, shutil

fiName = input("请输入需要遍历的文件夹地址:")
baseName = input("请输入需要查找文件的拓展名（如：.pdf,.txt,.jpg）:"