clojure - 使用 clojure 中的函数映射转换映射

Question

我正在尝试想出一种方法来转换 clojure HashMap ，方法是将来自另一个映射的函数应用于每个值。这是我到目前为止所拥有的:

(defn transform-map [m fm]
  (into {} (for [[k v] m]
    (let [mfn (get fm k identity)] [k (mfn v)])))

(def m {:a 0 :b 0 :c 0 :d 0})
(def fm {:a inc :b dec :c identity})
(transform-map m fm)  ;=> {:a 1, :b -1, :c 0, :d 0}

这很好用，但前提是每个函数都接受一个参数，即键的当前值。如果我想在我的函数映射中放置一个函数，该函数使用与同一键中的值不同的值怎么办？例如，假设我想将键 :a 和 :b 的和放入键 :d 中？

我可以尝试这样的事情:

(assoc fm :d (fn[a b] (+ a b)))

但有没有办法改变我的 transform-map 函数，以便它在该函数调用中使用适当的参数？

Answer 1

我建议分解函数以及它们如何适用于转换映射。下面是一个例子来说明:

;; functions which take some params and return a result
(defn sub [a b] (- a b))
(defn add [a b] (+ a b))

;; data map
(def data {:a 5 :b 4 :c 3})

;; transformation map key => output key, value is vector of function
;; to apply and the params that function will take from the data map
(def transformations {:a [inc :a]
                      :b [dec :b]
                      :c [add :a :b]
                      :d [sub :b :c]})

; The transformation function
(defn transform-map [m fm]
  (into {} (map (fn [[k v]]
                  [k (apply (first v)
                            ((apply juxt (rest v)) m))])
                fm)))

(transform-map data transformations)