将自己编的函数放入diy.py中:
# 判定路径下是否有文件夹,返回判断指标、文件夹数量和文件夹名称
def isincludedir(path): #输入路径
import os
dirs=os.listdir(path)
isincludedir_num=0
isincludedir_count=0
dir_includedir=[]
for dir in dirs:
pathson=path+"\\"+dir
if os.path.isdir(pathson):
isincludedir_num +=1
dir_includedir.append(dir)
isincludedir_count +=1
isincludedir=0 if isincludedir_num==0 else 1
return isincludedir,isincludedir_num,dir_includedir
# 罗列某一文件夹下后缀名为txt的文件名
def txtfile(path): #输入路径
import os
dirs=os.listdir(path)
txtfile_num=0
txtfile_count=0
dir_txtfile=[]
for dir in dirs:
pathson=path+"\\"+dir
if os.path.isfile(pathson) and os.path.splitext(dir)[1]==".txt":
dir_txtfile.append(dir)
return dir_txtfile
函数调用:
import sys
sys.path.insert(0, "G:")
import diy
print(diy.isincludedir("G:"))
print(diy.txtfile("G:"))