Write Clean Python Code Using Pipes

Motivation

map and filter are two efficient Python methods to work with iterables. However, the code can look messy if you use both map and filter at the same time.

Wouldn’t be nice if you can use pipes | to apply multiple methods on an iterable like below?

The library Pipe allows you to do exactly that.

Feel free to play and fork the source code of this article here.

What is Pipe?

Pipe is a Python library that enables you to use pipes in Python. A pipe (|) passes the results of one method to another method.

I like Pipe because it makes my code look cleaner when applying multiple methods to a Python iterable. Since Pipe only provides a few methods, it is also very easy to learn Pipe. In this article, I will show you some methods I found the most useful.

To install Pipe, type:

pip install pipe

Where — Filter Elements in an Iterable

Similar to SQL, Pipe’s where method can also be used to filter elements in an iterable.

Select — Apply a Function to an Iterable

The select method is similar to the map method. select applies a method to each element of an iterable.

In the code below, I use select to multiply each element in the list by 2.

Now, you might wonder: Why do we need the methods where and select if they have the same functionalities as map and filter ?

It is because you can insert one method after another method using pipes. As a result, using pipes removes nested parentheses and makes the code more readable.

Unfold Iterables

chain — Chain a Sequence of Iterables

It can be a pain to work with a nested iterable. Luckily, you can use chain to chain a sequence of iterables.

Even though the iterable is less nested after applying chain, we still have a nested list. To deal with a deeply nested list, we can use traverse instead.

traverse — Recursively Unfold Iterables

The traverse method can be used to recursively unfold iterables. Thus, you can use this method to turn a deeply nested list into a flat list.

Let’s integrate this method with the select method to get the values of a dictionary and flatten the list.

Pretty cool!

Group Elements in a List

Sometimes, it might be useful to group elements in a list using a certain function. That could be easily done with the groupby method.

To see how this method works, let’s turn a list of numbers into a dictionary that groups numbers based on whether they are even or odd.

In the code above, we use groupby to group numbers into theEven group and theOdd group. The output after applying this method looks like the below:

[('Even', <itertools._grouper at 0x7fbea8030550>),
 ('Odd', <itertools._grouper at 0x7fbea80309a0>)]

Next, we use select to turn a list of tuples into a list of dictionaries whose keys are the first elements in the tuples and values are the second elements in the tuples.

[{'Even': [2, 4, 6, 8]}, {'Odd': [1, 3, 5, 7, 9]}]

Cool! To get only the values that are greater than 2, we can add the where method inside the select method:

Note that there are no longer 2 and 1 in the outputs.

dedup — Deduplicate Values Using a Key

The dedup method removes duplicates in a list.

That might not sound interesting since the set method can do the same thing. However, this method is more flexible since it enables you to get unique elements using a key.

For example, you can use this method to get a unique element that is smaller than 5 and another unique element that is larger than or equal to 5.

Now, let’s combine this method with select and where to get the values of a dictionary that has duplicated keys and None values.

In the code above, we:

  • remove items with the same name
  • get the values of count
  • only choose the values that are integers.

Within a few lines of code, we can apply multiple methods to an iterable while still keeping the code clean. Pretty cool, isn’t it?

Conclusion

Congratulations! You have just learned how to use pipe to keep your code clean and short. I hope this article will give you the knowledge to turn complicated operations on an iterable into one simple line of code.

Scroll to Top