[英]What is the fastest way to check whether a directory is empty in Python
我在 Windows 机器上工作,想检查网络路径上的目录是否为空。
想到的第一件事是调用os.listdir()
并查看它的长度是否为 0。
IE
def dir_empty(dir_path):
return len(os.listdir(dir_path)) == 0
因为这是一个网络路径,我的连接并不总是很好,而且一个文件夹可能包含数千个文件,所以这是一个非常慢的解决方案。 有没有更好的?
迄今为止我找到的最快的解决方案:
def dir_empty(dir_path):
return not any((True for _ in os.scandir(dir_path)))
或者,如以下评论中所建议的:
def dir_empty(dir_path):
return not next(os.scandir(dir_path), None)
在我正在处理的慢速网络上,这需要几秒钟而不是几分钟(os.listdir() 版本是几分钟)。 这似乎更快,因为 any 语句只评估第一个 True 语句。
从 Python 3.4 开始,您可以使用pathlib.iterdir()
这将产生目录内容的路径对象:
>>> from pathlib import Path
>>>
>>> def dir_empty(dir_path):
... path = Path(dir_path)
... has_next = next(path.iterdir(), None)
... if has_next is None:
... return True
... return False
由于 OP 询问最快的方式,我认为使用os.scandir
并在我们发现第一个文件后立即返回应该是最快的。 os.scandir
返回一个迭代器。 我们应该避免创建一个完整的列表来检查它是否为空。
测试目录包含大约 10 万个文件:
from pathlib import Path
import os
path = 'jav/av'
len(os.listdir(path))
>>> 101204
然后开始我们的测试:
def check_empty_by_scandir(path):
with os.scandir(path) as it:
return not any(it)
def check_empty_by_listdir(path):
return not os.listdir(path)
def check_empty_by_pathlib(path):
return not any(Path(path).iterdir())
%timeit check_empty_by_scandir(path)
>>> 179 µs ± 878 ns per loop (mean ± std. dev. of 7 runs, 10000 loops each)
%timeit check_empty_by_listdir(path)
>>> 28 ms ± 185 µs per loop (mean ± std. dev. of 7 runs, 10 loops each)
%timeit check_empty_by_pathlib(path)
>>> 27.6 ms ± 140 µs per loop (mean ± std. dev. of 7 runs, 10 loops each)
如我们所见, check_empty_by_listdir
和check_empty_by_pathlib
比check_empty_by_scandir
慢约 155 倍。 os.listdir() 和 Path.iterdir() 的结果是相同的,因为 Path.iterdir() 在后台使用 os.listdir() ,在内存中创建了一个完整的列表。
此外,正如人们指出的那样,读取 os.stat 不是一种选择,它在 linux 中的空目录上返回 4096。
listdir
给出了一个列表。 scandir
给出了一个迭代器,它的性能可能更高。
def dir_empty(dir_path):
try:
next(os.scandir(dir_path))
return False
except StopIteration:
return True
在 Windows 操作系统上有PathIsDirectoryEmptyA 。 我们可以用它来检查文件夹是否为空。
def is_dir_empty(path:str)->bool:
import ctypes
shlwapi = ctypes.OleDLL('shlwapi')
return shlwapi.PathIsDirectoryEmptyA(path.encode('utf-8'))
使用os.stat
:
is_empty = os.stat(dir_path).st_size == 0
使用 Python 的pathlib :
from pathlib import Path
is_empty = Path(dir_path).stat().st_size == 0
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.