seq 上的滑动窗口

Question

在 Clojure 中，在（有限的，不是太大的）seq 上设置滑动窗口的最好方法是什么？我应该只使用 drop 和 take 并跟踪当前索引还是有更好的方法我错过了？

Answer 1

我认为步骤 1 的 partition 可以做到这一点：

user=> (partition 3 1 [3 1 4 1 5 9])
((3 1 4) (1 4 1) (4 1 5) (1 5 9))

Answer 2

如果你想在windows上操作，用map也可以很方便：

user=> (def a [3 1 4 1 5 9])
user=> (map (partial apply +) (partition 3 1 a))
(8 6 10 15)
user=> (map + a (next a) (nnext a))
(8 6 10 15)

Answer 3

我不知道 partition 可以做到这一点，所以我这样实现了

(defn sliding-window [seq length]
  (loop [result ()
         remaining seq]
    (let [chunk (take length remaining)]
      (if (< (count chunk) length)
        (reverse result)
        (recur (cons chunk result) (rest remaining))))))