Sliding window over seq
Asked Answered
B

3

19

In Clojure, what would be the nicest way to have a sliding window over a (finite, not too large) seq? Should I just use drop and take and keep track of the current index or is there a nicer way I'm missing?

Backrest answered 15/9, 2009 at 15:30 Comment(0)
O
33

I think that partition with step 1 does it:

user=> (partition 3 1 [3 1 4 1 5 9])
((3 1 4) (1 4 1) (4 1 5) (1 5 9))
Omnibus answered 15/9, 2009 at 17:5 Comment(0)
M
3

If you want to operate on the windows, it can also be convenient to do this with 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)
Martguerita answered 16/9, 2009 at 0:52 Comment(0)
V
1

I didn't know partition could do this so I implemented it this way

(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))))))
Vassaux answered 18/6, 2019 at 8:49 Comment(0)

© 2022 - 2024 — McMap. All rights reserved.