diff --git a/snippets/intersection_by.md b/snippets/intersection_by.md new file mode 100644 index 000000000..2601c36ec --- /dev/null +++ b/snippets/intersection_by.md @@ -0,0 +1,19 @@ +--- +title: intersection_by +tags: list,function,intermediate +--- + +Returns a list of elements that exist in both lists, after applying the provided function to each list element of both. + +Create a `set` by applying `fn` to each element in `b`, then use list comprehension in combination with `fn` on `a` to only keep values contained in both lists. + +```py +def intersection_by(a, b, fn): + _b = set(map(fn, b)) + return [item for item in a if fn(item) in _b] +``` + +```py +from math import floor +intersection_by([2.1, 1.2], [2.3, 3.4],floor) # [2.1] +``` \ No newline at end of file