不多说,引入os库
创建获取txt的函数,这一段是这段代码最关键的部分
用wolk返回所有文件,然后再通过循环查找到的文件,取出对应的后缀名做判断,相同的加入,不同的放弃
###获取指定目录下的txt文件名存于列表中的函数
def getTxt(dir):
L = []
###os.walk(file_dir) 返回参数说明:返回一个三元组 dirpath, dirnames, filenames,(文件夹下有文件夹和文件都可以)
for root, dirs, files in os.walk(dir):
for file in files:
####判断后缀是否为txt,想找其它的也可改成别的后缀比如图片.jpg
if os.path.splitext(file)[1] == '.txt':
L.append(os.path.join(file))
return L
调用函数并输出结果
print(getTxt('C:/Users/Administrator/Desktop/111/'))
源代码如下:
import os
###获取指定目录下的txt文件名存于列表中的函数
def getTxt(dir):
L = []
###os.walk(file_dir) 返回参数说明:返回一个三元组 dirpath, dirnames, filenames,(文件夹下有文件夹和文件都可以)
for root, dirs, files in os.walk(dir):
for file in files:
####判断后缀是否为txt,想找其它的也可改成别的后缀比如图片.jpg
if os.path.splitext(file)[1] == '.txt':
L.append(os.path.join(file))
return L
print(getTxt('C:/Users/Administrator/Desktop/111/'))