Python编程中可能经常用到的函数
1.os.walk()
一般用法为
import os ph=r'D:\temp\build' for root,dirs,files in os.walk(ph): print(root,dirs,files)
root指的是当前所在的文件夹路径,dirs是当前文件夹路径下的文件夹列表,files是当前文件夹路径下的文件列表。
2.filter(function, iterable)
该接收两个参数,第一个为函数,第二个为序列,序列的每个元素作为参数传递给函数进行判断,然后返回 True 或 False,最后将返回 True 的元素放到新列表中。
3.endswith()
用于判断字符串是否以指定后缀结尾,如果以指定后缀结尾返回True,否则返回False。可选参数"start"与"end"为检索字符串的开始与结束位置。
上述三种函数组合用于遍历目录下的文件
for root,dirs,_ in os.walk(data_dir): for sub_dir in dirs: img_names = os.listdir(os.path.join(root,sub_dir)) img_names = list(filter(lambda x:x.endswith('.jpg'),img_names))