List Difference Based on Function

PythonPythonBeginner
Practice Now

This tutorial is from open-source community. Access the source code

Introduction

In Python, we can easily find the difference between two lists using the built-in set() function. However, what if we want to find the difference based on a specific function applied to each element in both lists? In this challenge, you will create a function that takes in two lists and a function, and returns the difference between the two lists after applying the provided function to each list element of both.


Skills Graph

%%%%{init: {'theme':'neutral'}}%%%% flowchart RL python(("`Python`")) -.-> python/BasicConceptsGroup(["`Basic Concepts`"]) python(("`Python`")) -.-> python/ControlFlowGroup(["`Control Flow`"]) python(("`Python`")) -.-> python/DataStructuresGroup(["`Data Structures`"]) python(("`Python`")) -.-> python/FunctionsGroup(["`Functions`"]) python(("`Python`")) -.-> python/ModulesandPackagesGroup(["`Modules and Packages`"]) python(("`Python`")) -.-> python/PythonStandardLibraryGroup(["`Python Standard Library`"]) python/BasicConceptsGroup -.-> python/comments("`Comments`") python/BasicConceptsGroup -.-> python/variables_data_types("`Variables and Data Types`") python/ControlFlowGroup -.-> python/conditional_statements("`Conditional Statements`") python/ControlFlowGroup -.-> python/for_loops("`For Loops`") python/ControlFlowGroup -.-> python/list_comprehensions("`List Comprehensions`") python/DataStructuresGroup -.-> python/lists("`Lists`") python/DataStructuresGroup -.-> python/tuples("`Tuples`") python/DataStructuresGroup -.-> python/dictionaries("`Dictionaries`") python/FunctionsGroup -.-> python/function_definition("`Function Definition`") python/FunctionsGroup -.-> python/lambda_functions("`Lambda Functions`") python/ModulesandPackagesGroup -.-> python/importing_modules("`Importing Modules`") python/ModulesandPackagesGroup -.-> python/using_packages("`Using Packages`") python/ModulesandPackagesGroup -.-> python/standard_libraries("`Common Standard Libraries`") python/PythonStandardLibraryGroup -.-> python/math_random("`Math and Random`") python/PythonStandardLibraryGroup -.-> python/data_collections("`Data Collections`") python/FunctionsGroup -.-> python/build_in_functions("`Build-in Functions`") subgraph Lab Skills python/comments -.-> lab-13621{{"`List Difference Based on Function`"}} python/variables_data_types -.-> lab-13621{{"`List Difference Based on Function`"}} python/conditional_statements -.-> lab-13621{{"`List Difference Based on Function`"}} python/for_loops -.-> lab-13621{{"`List Difference Based on Function`"}} python/list_comprehensions -.-> lab-13621{{"`List Difference Based on Function`"}} python/lists -.-> lab-13621{{"`List Difference Based on Function`"}} python/tuples -.-> lab-13621{{"`List Difference Based on Function`"}} python/dictionaries -.-> lab-13621{{"`List Difference Based on Function`"}} python/function_definition -.-> lab-13621{{"`List Difference Based on Function`"}} python/lambda_functions -.-> lab-13621{{"`List Difference Based on Function`"}} python/importing_modules -.-> lab-13621{{"`List Difference Based on Function`"}} python/using_packages -.-> lab-13621{{"`List Difference Based on Function`"}} python/standard_libraries -.-> lab-13621{{"`List Difference Based on Function`"}} python/math_random -.-> lab-13621{{"`List Difference Based on Function`"}} python/data_collections -.-> lab-13621{{"`List Difference Based on Function`"}} python/build_in_functions -.-> lab-13621{{"`List Difference Based on Function`"}} end

List Difference Based on Function

Create a function called difference_by(a, b, fn) that takes in three parameters:

  • a: a list of elements
  • b: a list of elements
  • fn: a function that will be applied to each element in both lists

The function should return a list of elements that are in list a but not in list b, after applying the provided function fn to each element in both lists.

To solve this problem, you can follow these steps:

  1. Create a set, using map() to apply fn to each element in b.
  2. Use a list comprehension in combination with fn on a 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 } ]

Summary

In this challenge, you have learned how to find the difference between two lists based on a specific function applied to each element in both lists. You have also practiced using map(), set(), and list comprehension to solve the problem.

Other Python Tutorials you may like