Trong clojure, làm thế nào để hợp nhất một số bản đồ kết hợp ánh xạ với cùng một khóa vào một danh sách?

Question

Trong Clojure, tôi muốn kết hợp nhiều bản đồ vào một bản đồ duy nhất nơi ánh xạ với cùng một khóa được kết hợp thành một danh sách.

Ví dụ:

{:humor :happy} {:humor :sad} {:humor :happy} {:weather :sunny} 

sẽ dẫn đến:

{:weather :sunny, :humor (:happy :sad :happy)} 

Tôi nghĩ về:

(merge-with (comp flatten list) data) 

Nhưng nó không phải là hiệu quả vì flatten có O (n) phức tạp.

Sau đó, tôi đã đưa ra:

(defn agg[x y] (if (coll? x) (cons y x) (list y x))) 
(merge-with agg data) 

Nhưng nó không cảm thấy ngữ. Bạn có ý tưởng nào khác không?

Answer 1

Một cách tiếp cận sẽ

(defn merge-lists [& maps] 
    (reduce (fn [m1 m2] 
      (reduce (fn [m [k v]] 
         (update-in m [k] (fnil conj []) v)) 
        m1, m2)) 
      {} 
      maps)) 

Đó là một chút xấu xí, nhưng đó là chỉ vì giá trị của bạn chưa được liệt kê. Nó cũng buộc mọi thứ làm danh sách (vì vậy bạn sẽ nhận được :weather [:sunny] thay vì :weather :sunny). Thành thật mà nói, điều này có thể là một triệu lần dễ dàng hơn cho bạn để làm việc với anyway.

Nếu bạn có từng giá trị dưới dạng véc tơ, bạn có thể chỉ cần thực hiện (apply merge-with into maps).

Answer 2

Bạn có thể thử những điều sau đây, tôi nghĩ rằng nó khá hiệu quả

(reduce 
    (fn [m pair] (let [[[k v]] (seq pair)] 
       (assoc m k (cons v (m k))))) 
    {} 
    data) 

=> {:weather (:sunny), :humor (:happy :sad :happy)} 

Answer 3

câu trả lời của @ amalloy có thể làm phẳng một chút bằng cách sử dụng for.

(reduce (fn [m [k v]] (update-in m [k] (fnil conj []) v)) {} (for [m data entry m] entry)) 

Nguồn cho kỹ thuật này: http://clj-me.cgrand.net/2010/01/19/clojure-refactoring-flattening-reduces/

Answer 4

Dưới đây là một giải pháp mà mỗi giá trị được biểu diễn dưới dạng danh sách, ngay cả khi độc thân:

(->> [{:humor :happy} {:humor :sad} {:humor :happy} {:weather :sunny}] 
    (map first) 
    (reduce (fn [m [k v]] (update-in m [k] #(cons v %))) {})) 

=> {:weather (:sunny), :humor (:happy :sad :happy)} 

Nếu bạn không muốn quấn độc thân trong một danh sách sau đó tôi nghĩ rằng giải pháp ban đầu của bạn là tốt. Cách duy nhất để làm cho nó thành ngữ hơn là sử dụng core.match.

(->> [{:humor :happy} {:humor :sad} {:humor :happy} {:weather :sunny}] 
    (apply merge-with #(match %1 
           [& _] (conj %1 %2) 
           :else [%1 %2]))) 

=> {:weather :sunny, :humor [:happy :sad :happy]} 

Answer 5

Merge với chức năng này:

(defn acc-list [x y] 
    (let [xs (if (seq? x) x (cons x nil))] 
    (cons y xs))) 

Answer 6

gì về việc sử dụng nhóm-by?Nó không trả lại chính xác những gì bạn yêu cầu nhưng nó là rất giống nhau:

user=> (group-by first (concat {:humor :happy} {:humor :sad} {:humor :happy} {:weather :sunny :humor :whooot})) 
{:humor [[:humor :happy] [:humor :sad] [:humor :happy] [:humor :whooot]], :weather [[:weather :sunny]]} 

Hoặc với một thay đổi nhỏ để nhóm theo chức năng:

(defn group-by-v2 
[f vf coll] 
    (persistent! 
    (reduce 
    (fn [ret x] 
     (let [k (f x)] 
     (assoc! ret k (conj (get ret k []) (vf x))))) 
    (transient {}) coll))) 

trở thành:

user=> (group-by-v2 key val (concat {:humor :happy} {:humor :sad} {:humor :happy} {:weather :sunny :humor :whooot})) 
{:humor [:happy :sad :happy :whooot], :weather [:sunny]}