dictionary - Clojure - 沿着路径行走

Question

我正在寻找类似于 clojure.walk 中的功能有 inner作为参数的函数:

不是键和值，就像 clojure.walk/walk 函数

但是从顶级数据结构访问值所需的键向量。

递归遍历所有数据

例子 :

;; not good since it takes `[k v]` as argument instead of `[path v]`, and is not recursive.
user=> (clojure.walk/walk (fn [[k v]] [k (* 10 v)]) identity {:a 1 :b {:c 2}})
;; {:a 10, :c 30, :b 20}

;; it should receive as arguments instead :
[[:a] 1]
[[:b :c] 2]

笔记:

它也应该使用数组，使用键 0、1、2...(就像在 get-in 中一样)。

我真的不在乎 outer参数，如果这允许简化代码。

Answer 1

目前正在学习 clojure，我尝试将此作为练习。
然而，我发现直接实现它是非常棘手的，就像沿着树向下走一样应用内部函数。

为了实现您正在寻找的结果，我将任务分为 2 个:

首先将嵌套结构转换为以路径为键，值为

的字典。

然后映射内部函数，或者用外部函数减少。

我的实现:

;; Helper function to have vector's indexes work like for get-in
(defn- to-indexed-seqs [coll]
  (if (map? coll)
    coll
    (map vector (range) coll)))

;; Flattening the tree to a dict of (path, value) pairs that I can map over
;; user>  (flatten-path [] {:a {:k1 1 :k2 2} :b [1 2 3]})
;; {[:a :k1] 1, [:a :k2] 2, [:b 0] 1, [:b 1] 2, [:b 2] 3}
(defn- flatten-path [path step]
  (if (coll? step)
    (->> step
         to-indexed-seqs
         (map (fn [[k v]] (flatten-path (conj path k) v)))
         (into {}))
    [path step]))

;; Some final glue
(defn path-walk [f coll]
  (->> coll
      (flatten-path [])
      (map #(apply f %))))

;; user> (println (clojure.string/join "\n" (path-walk #(str %1 " - " %2) {:a {:k1 1 :k2 2} :b [1 2 3]})))
;; [:a :k1] - 1
;; [:a :k2] - 2
;; [:b 0] - 1
;; [:b 1] - 2
;; [:b 2] - 3