Merge pull request #39 from aollier/intersect

intersect : une version courte et efficace
flotpython · Nov 21, 2020 · 7ac5daa · 7ac5daa
2 parents 589449a + 25d251f
commit 7ac5daa
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/modules/corrections/exo_intersect.py b/modules/corrections/exo_intersect.py
@@ -34,6 +34,13 @@ def keys(tuples):
          | {val_b for key, val_b in tuples_b if key in common_keys}
 # @END@
 
+# @BEG@ name=intersect more=bis
+def intersect_bis(A, B):
+    A, B = dict(A), dict(B)
+    keys = set(A) & set(B)
+    return {A[k] for k in keys} | {B[k] for k in keys}
+# @END@
+
 
 def intersect_ko(A, B):
     A_vals = { v for k, v in A }