Clojure 减少包含混合向量和双精度向量的函数

Question

I would like sum a vector containing a mixture of doubles and vectors. Something like,

[[1 2 1 [1 2 3]] [1 2 4 [1 1 1]]...]

That I would like to sum such that I get something like,

[212 12 444 [11 2 12]]

Is there an efficient way to do this using core clojure functions like reduce or map?

Answer 1

I think that you want to have a function that adds a nested vector structure, so that (nvadd [1 2 1 [1 2 3]] [1 2 4 [1 1 1]]) evaluates to [2 4 5 [2 3 4]] . You can then reduce your vector of such vectors with that function.

(defn nvadd [a b]
  (mapv #(if (vector? %1)
           (nvadd %1 %2)
           (+ %1 %2))
        a b))

This assumes well-formed input.