在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
函数,然后打印出所有的滑动窗口。
原创文章,作者:未希,如若转载,请注明出处:https://www.kdun.com/ask/452908.html
本网站发布或转载的文章及图片均来自网络,其原创性以及文中表达的观点和判断不代表本网站。如有问题,请联系客服处理。
发表回复