上一篇
如何用python实现滑动窗口
- 行业动态
- 2024-04-08
- 2638
在Python中,我们可以使用内置的collections模块中的deque数据结构来实现滑动窗口,以下是一个简单的例子:
from collections import deque def sliding_window(iterable, n): # 创建一个双端队列 dq = deque(maxlen=n) for item in iterable: dq.append(item) if len(dq) == n: yield list(dq) 测试代码 for window in sliding_window([1, 2, 3, 4, 5], 3): print(window)
在这个例子中,我们首先从collections模块导入deque类,我们定义了一个名为sliding_window的函数,它接受一个可迭代对象和一个窗口大小作为参数。
在函数内部,我们创建了一个最大长度为n的双端队列dq,我们遍历输入的可迭代对象,将每个元素添加到队列的末尾,当队列的长度达到n时,我们就生成一个新的列表,其中包含队列中的所有元素。
我们使用一个简单的循环来测试我们的函数,我们将一个包含五个元素的列表和窗口大小3传递给sliding_window函数,然后打印出所有的滑动窗口。
本站发布或转载的文章及图片均来自网络,其原创性以及文中表达的观点和判断不代表本站,有问题联系侵删!
本文链接:http://www.xixizhuji.com/fuzhu/320396.html