How To Use the Python Map Function

Ultimately map() is most useful when working with large datasets, so we would likely work with the map object further, and generally would not be using a constructor like list() on them.

For smaller datasets, list comprehensions may be more suitable, but for the purposes of this tutorial we’re using a small dataset to demonstrate map().

Similarly to a lambda we can use a function we have defined to apply to an iterable. While lambda functions are more useful to implement when you’re working with a one-line expression, user-defined functions are more appropriate when the expression grows in complexity. Furthermore, when we need to pass another piece of data to the function that you’re applying to your iterable, user-defined functions can be a better choice for readability.

For example, in the following iterable, each item is a dictionary that contains different details about each of our aquarium creatures:

aquarium_creatures = [ {“name”: “sammy”, “species”: “shark”, “tank number”: 11, “type”: “fish”}, {“name”: “ashley”, “species”: “crab”, “tank number”: 25, “type”: “shellfish”}, {“name”: “jo”, “species”: “guppy”, “tank number”: 18, “type”: “fish”}, {“name”: “jackie”, “species”: “lobster”, “tank number”: 21, “type”: “shellfish”}, {“name”: “charlie”, “species”: “clownfish”, “tank number”: 12, “type”: “fish”}, {“name”: “olly”, “species”: “green turtle”, “tank number”: 34, “type”: “turtle”} ]

We’ve decided that all the aquarium creatures are in fact going to move into the same tank. We need to update our records to reflect that all of our creatures are moving into tank 42. To have map() access each dictionary and each key:value pair in the dictionaries, we construct a nested function:

def assign_to_tank(aquarium_creatures, new_tank_number): def apply(x): x[“tank number”] = new_tank_number return x return map(apply, aquarium_creatures)

We define an assign_to_tank() function that takes aquarium_creatures and new_tank_number as parameters. In assign_to_tank() we pass apply() as the function to map() on the final line. The assign_to_tank function will return the iterator resulting from map().

apply() takes x as an argument, which represents an item in our list — a single dictionary.

Next we define that x is the “tank number” key from aquarium_creatures and that it should store the passed in new_tank_number. We return each item after applying the new tank number.

We call assign_to_tank() with our list of dictionaries and the new tank number we want to replace for each creature:

assigned_tanks = assign_to_tank(aquarium_creatures, 42)

Once the function completes we have our map object stored in the assigned_tanks variable, which we turn into a list and print:

print(list(assigned_tanks))

We’ll receive the following output from this program:

Output

We’ve mapped the new tank number to our list of dictionaries. Using a function that we define, we can incorporate map() to apply the function efficiently on each item of the list.

In the same way as lambda functions or our own defined functions, we can use Python built-in functions with map(). To apply a function with multiple iterables, we pass in another iterable name following the first one. For example, using the pow() function that takes in two numbers to find the power of the base number to the provided exponent.

Here we have our lists of integers that we would like to use with pow():

base_numbers = [2, 4, 6, 8, 10] powers = [1, 2, 3, 4, 5]

Next we pass in pow() as our function into map() and provide the two lists as our iterables:

numbers_powers = list(map(pow, base_numbers, powers)) print(numbers_powers)

map() will apply the pow() function to the same item in each list to provide the power. Therefore our results will show 2**1, 4**2, 6**3, and so on:

Output

If we were to provide map() with an iterable that was longer than the other, map() would stop calculating once it reaches the end of the shortest iterable. In the following program we’re extending base_numbers with three additional numbers:

base_numbers = [2, 4, 6, 8, 10, 12, 14, 16] powers = [1, 2, 3, 4, 5] numbers_powers = list(map(pow, base_numbers, powers)) print(numbers_powers)

As a result, nothing will change within the calculation of this program and so it will still yield the same result:

Output

We’ve used the map() function with a Python built-in function and have seen that it can handle multiple iterables. We’ve also reviewed that map() will continue to process multiple iterables until it has reached the end of the iterable with the fewest items.

In this tutorial, we’ve learned the different ways of using the map() function in Python. Now you can use map() with your own function, a lambda function, and with any other built-in functions. You can also implement map() with functions that require multiple iterables.

In this tutorial, we printed the results from map() immediately to a list format for demonstration purposes. In our programs we would typically use the returned map object to further manipulate the data.

If you would like to learn more Python, check out our How To Code in Python 3 series and our Python topic page. To learn more about working with data sets in functional programming, check out our article on the filter() function.