如何用python实现滑动窗口

在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

本网站发布或转载的文章及图片均来自网络，其原创性以及文中表达的观点和判断不代表本网站。如有问题，请联系客服处理。