如何判断文件夹是否包含具有特定扩展名的文件？

Question

给定一个文件夹，如何判断它是否包含指定扩展名的任何文件？我不需要文件名或包含该扩展名的文件数量;我只需要确定是否存在带扩展名的文件。

Answer 1

使用any函数测试任何文件字符串是否以所需的扩展名字符串结尾。

any(i.endswith(extension) for i in os.listdir(directory))

Answer 2

这个简单的单行将会这样做：

any(ext == os.path.splitext(name)[1] for name in os.listdir(path))

假设path是dir路径，ext是扩展字符串。