查找目录下匹配.py或者.txt的文件

#!/usr/bin/python

def endWith(s,*endstring_1):       #*代表允许传输多个参数，名称统一为endstring为元祖，**为字典

array = map(s.endswith,endstring_1)  #map(调用的函数名称，给这个函数传的参数)

if True in array:

return True

else:

return False

if __name__ == ‘__main__‘:

import os

s = os.listdir(‘/home/yanchao/‘)

f_file = []

for i in s:

if endWith(i,‘.txt‘,‘.py‘):

print i,

查找目录下匹配.py或者.txt的文件