Home > Dive Into Python > Data-Centric Programming > Mapping lists revisited | << >> | ||||
diveintopython.org Python for experienced programmers |
You're already familiar with using list comprehensions to map one list into another. There is another way to accomplish the same thing, using the built-in map function. It works much the same way as the filter function.
>>> def double(n): ... return n*2 ... >>> li = [1, 2, 3, 5, 9, 10, 256, -3] >>> map(double, li) [2, 4, 6, 10, 18, 20, 512, -6] >>> [double(n) for n in li] [2, 4, 6, 10, 18, 20, 512, -6] >>> newlist = [] >>> for n in li: ... newlist.append(double(n)) ... >>> newlist [2, 4, 6, 10, 18, 20, 512, -6]
map takes a function and a list[15] and returns a new list by calling the function with each element of the list in order. In this case, the function simply multiplies each element by 2. | |
You could accomplish the same thing with a list comprehension. List comprehensions were first introduced in Python 2.0; map has been around forever. | |
You could, if you insist on thinking like a Visual Basic programmer, use a for loop to accomplish the same thing. |
Example 7.11. map with lists of mixed datatypes
>>> li = [5, 'a', (2, 'b')] >>> map(double, li) [10, 'aa', (2, 'b', 2, 'b')]
All right, enough play time. Let's look at some real code.
Example 7.12. map in regression.py
filenameToModuleName = lambda f: os.path.splitext(f)[0] moduleNames = map(filenameToModuleName, files)
As we saw in Using lambda functions, lambda defines an inline function. And as we saw in Example 3.36, os.path.splitext takes a filename and returns a tuple (name, extension). So filenameToModuleName is a function which will take a filename and strip off the file extension, and return just the name. | |
Calling map takes each filename listed in files, passes it to our function filenameToModuleName, and returns a list of the return values of each of those function calls. In other words, we strip the file extension off of each filename, and store the list of all those stripped filenames in moduleNames. |
As we'll see in the rest of the chapter, we can extend this type of data-centric thinking all the way to our final goal, which is to define and execute a single test suite that contains the tests from all of those individual test suites.
Footnotes
[15] Again, I should point out that map can take a list, a tuple, or any object that acts like a sequence. See previous footnote about filter.
Filtering lists revisited | 1 2 3 4 5 6 | Data-centric programming |