Skip to content

List intersection based on function

Returns a list of elements that exist in both lists, after applying the provided function to each list element of both.

Create a set, using map() to apply fn to each element in b.
Use a list comprehension in combination with fn on a to only keep values contained in both lists.

def intersection_by(a, b, fn):
  _b = set(map(fn, b))
  return [item for item in a if fn(item) in _b]

from math import floor

intersection_by([2.1, 1.2], [2.3, 3.4], floor) # [2.1]

More like this

Collection · 11 snippets
Python Functions

A snippet collection of function helpers and tips for Python 3.6.
Python · November 2, 2020
List union based on function

Returns every element that exists in any of the two lists once, after applying the provided function to each element of both.
Python · November 2, 2020
List difference based on function

Returns the difference between two lists, after applying the provided function to each list element of both.
Python · November 2, 2020
List symmetric difference based on function

Returns the symmetric difference between two lists, after applying the provided function to each list element of both.