blob: 6b54dd913692fd3aa89f5c87a08dcdd98b3316ad (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
|
; Copyright (c) Rich Hickey. All rights reserved.
; The use and distribution terms for this software are covered by the
; Common Public License 1.0 (http://opensource.org/licenses/cpl.php)
; which can be found in the file CPL.TXT at the root of this distribution.
; By using this software in any fashion, you are agreeing to be bound by
; the terms of this license.
; You must not remove this notice, or any other, from this software.
(in-ns 'set)
(clojure/refer 'clojure)
(defn
#^{:doc "Returns a set that is the union of the two sets."}
union [xset yset]
(reduce conj xset yset))
(defn
#^{:doc "Returns a set that is xset without the elements of yset."}
difference [xset yset]
(reduce disj xset yset))
(defn
#^{:doc "Returns a set of the elements present in both xset and yset."}
intersection [xset yset]
(difference xset (difference xset yset)))
(defn
#^{:doc "Returns a set of the elements for which pred is true"}
select [pred xset]
(reduce (fn [s k] (if (pred k) (disj s k) s))
xset xset))
(defn
#^{:doc "Returns a rel of the elements of xrel with only the keys in ks"}
project [xrel ks]
(set (map #(select-keys % ks) xrel)))
(defn
#^{:doc "Returns the map with the keys in kmap renamed to the vals in kmap"}
rename-keys [map kmap]
(reduce
(fn [m [old new]]
(if (not= old new)
(-> m (assoc new (m old)) (dissoc old))
m))
map kmap))
(defn
#^{:doc "Returns a rel of the maps in xrel with the keys in kmap renamed to the vals in kmap"}
rename [xrel kmap]
(set (map #(rename-keys % kmap) xrel)))
(defn
#^{:doc "Returns a map of the distinct values of ks in the xrel mapped to a
set of the maps in xrel with the corresponding values of ks."}
index [xrel ks]
(reduce
(fn [m x]
(let [ik (select-keys x ks)]
(assoc m ik (conj (get m ik #{}) x))))
{} xrel))
(defn
#^{:doc "Returns the map with the vals mapped to the keys."}
map-invert [m]
(reduce (fn [m [k v]] (assoc m v k)) {} m))
(defn
#^{:doc "When passed 2 rels, returns the rel corresponding to the natural join.
When passed an additional keymap, joins on the corresponding keys."}
join
([xrel yrel] ;natural join
(when (and (seq xrel) (seq yrel))
(let [ks (intersection (set (keys (first xrel))) (set (keys (first yrel))))
[r s] (if (<= (count xrel) (count yrel))
[xrel yrel]
[yrel xrel])
idx (index r ks)]
(reduce (fn [ret x]
(let [found (idx (select-keys x ks))]
(if found
(reduce #(conj %1 (merge %2 x)) ret found)
ret)))
#{} s))))
([xrel yrel km] ;arbitrary key mapping
(let [[r s k] (if (<= (count xrel) (count yrel))
[xrel yrel (map-invert km)]
[yrel xrel km])
idx (index r (vals k))]
(reduce (fn [ret x]
(let [found (idx (rename-keys (select-keys x (keys k)) k))]
(if found
(reduce #(conj %1 (merge %2 x)) ret found)
ret)))
#{} s))))
(comment
(refer 'set)
(def xs #{{:a 11 :b 1 :c 1 :d 4}
{:a 2 :b 12 :c 2 :d 6}
{:a 3 :b 3 :c 3 :d 8 :f 42}})
(def ys #{{:a 11 :b 11 :c 11 :e 5}
{:a 12 :b 11 :c 12 :e 3}
{:a 3 :b 3 :c 3 :e 7 }})
(join xs ys)
(join xs (rename ys {:b :yb :c :yc}) {:a :a})
(union #{:a :b :c} #{:c :d :e })
(difference #{:a :b :c} #{:c :d :e})
(intersection #{:a :b :c} #{:c :d :e})
(index ys [:b])
)
|
