Python：如何获取目录中的第一个文件？

Question

所以我想在Python中抓取目录下的第一个文件。我知道我可以这样做：

first_file = [join(path, f) for f in os.listdir(path) if isfile(join(path, f))][0]

但它很慢。有没有更好的解决办法？谢谢！

Answer 1

您可以使用next()：

first_file = next(join(path, f) for f in os.listdir(path) if isfile(join(path, f)))

请注意，如果目录中没有文件，它将抛出StopIteration异常。要么处理它，要么提供一个默认值：

first_file = next((join(path, f) for f in os.listdir(path) if isfile(join(path, f))), 
                  "default value here")