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))