python：调用自己编的函数

将自己编的函数放入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:"))