python获取文件夹里有什么文件+查看特定格式的文件

功能

获取文件夹的的文件+获取文件夹里的特定格式的文件，比如.png等

程序

import  os
folder = r"D:\python\untitled1"
files = os.listdir(folder)
print(files)

def file_filter(f):
    if f[-4:] in ['.jpg', '.png']:
        return True
    else:
        return False

files = list(filter(file_filter, files))
print(files)

效果

想要查看的文件夹

运行程序

可以看到，首先打印出所有格式的文件名，之后打印我们锁定的格式的文件名。

后续

如果想了解更多物联网、智能家居项目知识，可以关注我的程序设计专栏。
或者关注公众号。

编写不易，感谢支持。