Returns the difference between two lists, after applying the provided function to each list element of both.
- Create a
set
, usingmap()
to applyfn
to each element inb
. - Use a list comprehension in combination with
fn
ona
to only keep values not contained in the previously created set,_b
.
代码实现
def difference_by(a, b, fn):
_b = set(map(fn, b))
return [item for item in a if fn(item) not in _b]
使用样例
from math import floor
difference_by([2.1, 1.2], [2.3, 3.4], floor) # [1.2]
difference_by([{ 'x': 2 }, { 'x': 1 }], [{ 'x': 1 }], lambda v : v['x'])
# [ { x: 2 } ]