#获取某文件夹下制定类型文件
# import os
# def filep(fp):
# l=[]
# a=os.walk(fp) #生成器
# for nowp,sonp,oth in a: #当前目录,子文件夹(一级),其他文件费文件夹
# for i in oth:
# if os.path.splitext(i)[1]=='.py':#splitext分离路径和文件名
# l.append(i)
# return l
# print(filep("D:\\1"))
#================================
import os
def filen(ptt):
ll=[]
a=os.listdir(ptt) #返回的是当前路径下所有文件名称,包括文件夹的名称
print(a)
for i in a:
if os.path.splitext(i)[1]=='.py':
ll.append(i)
# filepath=os.path.join(ptt, i) #路径\名称
# if os.path.isdir(filepath): #判断是否为路径
# print(i)
return ll
print(filen("D:\\1"))