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

 

posted @ 2021-03-01 13:15  Anm半夏  阅读(69)  评论(0编辑  收藏  举报