Files
30-seconds-of-code/python/s/intersection-by.md
Angelos Chalaris 5c913d20bd Reorganize snippets
2023-05-03 21:19:02 +03:00

662 B

title, type, language, tags, cover, dateModified
title type language tags cover dateModified
List intersection based on function snippet python
list
function
duck-plants 2020-11-02T19:28:05+02:00

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]