可以使用assoc-in创建新的向量而不是新的向量吗?

时间:2012-11-02 15:06:49

标签: clojure

assoc-in可以更改矢量/贴图中索引/键的值。在地图中,如果某个键不存在,则它会生成一个新的键/值对。是否存在类似于向量的内容(如果索引不存在,则在该索引处生成具有该值的列表)类似于:

 (reduce (fn [g [x y]] (assoc-in g [x y] y ))
        []
        (for [x (range 2) 
              y (range 2)]
             [x y]))

上面的代码生成:

[{1 1, 0 0} {1 1, 0 0}]

我希望它能够生成:

[[0 1] [0 1]]

这是否可以通过简单的方式实现?

由于

编辑: 更清楚一点,我只是希望它生成嵌套向量而不是嵌套映射(或映射向量)现在我把y作为值,但这只是一个例子。

2 个答案:

答案 0 :(得分:6)

在一般情况下,创建向量而不是地图不会很好,因为您只能assoc现有索引或向量中的下一个索引:

user=> (assoc [:a] 0 :b) ; works because index 0 exists
[:b]
user=> (assoc [:a] 1 :b) ; works because index 1 is next
[:a :b]
user=> (assoc [:a] 2 :b) ; fails since index is out of range
IndexOutOfBoundsException   clojure.lang.PersistentVector.assocN (PersistentVector.java:136)

但是,如果你小心,你仍然可以使它工作。以下是assoc-in的实现:

(defn assoc-in
  [m [k & ks] v]
  (if ks
    (assoc m k (assoc-in (get m k) ks v))
    (assoc m k v)))

请注意,它在get条件的真分支上调用if。如果您希望自己的assoc-in版本使用向量,则可以添加空向量作为get返回的默认值:

(defn vassoc-in
  [m [k & ks] v]
  (if ks
    (assoc m k (vassoc-in (get m k []) ks v))
    (assoc m k v)))

在您的问题的示例代码中使用vassoc-in而不是assoc-in会导致[[0 1] [0 1]],就像您想要的那样。

答案 1 :(得分:0)

您似乎需要以下内容:

user=> (let [cv #(apply vector %)] 
         (cv (map cv (repeat 5 (range 2)))))
[[0 1] [0 1] [0 1] [0 1] [0 1]]