Module enabling a sh like infix syntax (using pipes).
As an example, here is the solution for the 2nd Euler Project problem:
Find the sum of all the even-valued terms in Fibonacci which do not exceed four million.
Given fib
a generator of Fibonacci numbers:
sum(fib() | where(lambda x: x % 2 == 0) | take_while(lambda x: x < 4000000))
Each pipes is lazy evalatated, can be aliased, and partially initialized, so it could be rewritten as:
is_even = where(lambda x: x % 2 == 0)
sum(fib() | is_even | take_while(lambda x: x < 4000000)
To install the library, you can just run the following command:
# Linux/macOS
python3 -m pip install pipe
# Windows
py -3 -m pip install pipe
The basic syntax is to use a |
like in a shell:
>>> from itertools import count
>>> from pipe import select, take
>>> sum(count() | select(lambda x: x ** 2) | take(10))
285
>>>
Some pipes take an argument:
>>> from pipe import where
>>> sum([1, 2, 3, 4] | where(lambda x: x % 2 == 0))
6
>>>
Some do not need one:
>>> from pipe import traverse
>>> for i in [1, [2, 3], 4] | traverse:
... print(i)
1
2
3
4
>>>
In which case it's allowed to use the calling parenthesis:
>>> from pipe import traverse
>>> for i in [1, [2, 3], 4] | traverse():
... print(i)
1
2
3
4
>>>
Alphabetical list of available pipes; when several names are listed for a given pipe, these are aliases.
Like Python 3.12 itertool.batched
:
>>> from pipe import batched
>>> list("ABCDEFG" | batched(3))
[('A', 'B', 'C'), ('D', 'E', 'F'), ('G',)]
>>>
Chain a sequence of iterables:
>>> from pipe import chain
>>> list([[1, 2], [3, 4], [5]] | chain)
[1, 2, 3, 4, 5]
>>>
Warning : chain only unfolds an iterable containing ONLY iterables:
list([1, 2, [3]] | chain)
Gives a TypeError: 'int' object is not iterable
Consider using traverse.
Like itertools.chain, yields elements of the given iterable, then yields elements of its parameters
>>> from pipe import chain_with
>>> list((1, 2, 3) | chain_with([4, 5], [6]))
[1, 2, 3, 4, 5, 6]
>>>
Deduplicate values, using the given key
function if provided.
>>> from pipe import dedup
>>> list([-1, 0, 0, 0, 1, 2, 3] | dedup)
[-1, 0, 1, 2, 3]
>>> list([-1, 0, 0, 0, 1, 2, 3] | dedup(key=abs))
[-1, 0, 2, 3]
>>>
The builtin enumerate()
as a Pipe:
>>> from pipe import enumerate
>>> list(['apple', 'banana', 'citron'] | enumerate)
[(0, 'apple'), (1, 'banana'), (2, 'citron')]
>>> list(['car', 'truck', 'motorcycle', 'bus', 'train'] | enumerate(start=6))
[(6, 'car'), (7, 'truck'), (8, 'motorcycle'), (9, 'bus'), (10, 'train')]
>>>
Alias for where(predicate)
, see where(predicate)
.
Like itertools.groupby(sorted(iterable, key = keyfunc), keyfunc)
>>> from pipe import groupby, map
>>> items = range(10)
>>> ' / '.join(items | groupby(lambda x: "Odd" if x % 2 else "Even")
... | select(lambda x: "{}: {}".format(x[0], ', '.join(x[1] | map(str)))))
'Even: 0, 2, 4, 6, 8 / Odd: 1, 3, 5, 7, 9'
>>>
Just the itertools.islice
function as a Pipe:
>>> from pipe import islice
>>> list((1, 2, 3, 4, 5, 6, 7, 8, 9) | islice(2, 8, 2))
[3, 5, 7]
>>>
Just the itertools.izip
function as a Pipe:
>>> from pipe import izip
>>> list(range(0, 10) | izip(range(1, 11)))
[(0, 1), (1, 2), (2, 3), (3, 4), (4, 5), (5, 6), (6, 7), (7, 8), (8, 9), (9, 10)]
>>>
Apply a conversion expression given as parameter to each element of the given iterable
>>> list([1, 2, 3] | map(lambda x: x * x))
[1, 4, 9]
>>> list([1, 2, 3] | select(lambda x: x * x))
[1, 4, 9]
>>>
The netcat Pipe sends and receive bytes over TCP:
data = [
b"HEAD / HTTP/1.0\r\n",
b"Host: python.org\r\n",
b"\r\n",
]
for packet in data | netcat("python.org", 80):
print(packet.decode("UTF-8"))
Gives:
HTTP/1.1 301 Moved Permanently
Content-length: 0
Location: https://python.org/
Connection: close
Returns all possible permutations:
>>> from pipe import permutations
>>> for item in 'ABC' | permutations(2):
... print(item)
('A', 'B')
('A', 'C')
('B', 'A')
('B', 'C')
('C', 'A')
('C', 'B')
>>>
>>> for item in range(3) | permutations:
... print(item)
(0, 1, 2)
(0, 2, 1)
(1, 0, 2)
(1, 2, 0)
(2, 0, 1)
(2, 1, 0)
>>>
Like Python's built-in reversed
function.
>>> from pipe import reverse
>>> list([1, 2, 3] | reverse)
[3, 2, 1]
>>>
Alias for map(fct)
, see map(fct)
.
Skips the given quantity of elements from the given iterable, then yields
>>> from pipe import skip
>>> list((1, 2, 3, 4, 5) | skip(2))
[3, 4, 5]
>>>
Like itertools.dropwhile, skips elements of the given iterable while the predicate is true, then yields others:
>>> from pipe import skip_while
>>> list([1, 2, 3, 4] | skip_while(lambda x: x < 3))
[3, 4]
>>>
Like Python's built-in "sorted" primitive.
>>> from pipe import sort
>>> ''.join("python" | sort)
'hnopty'
>>> [5, -4, 3, -2, 1] | sort(key=abs)
[1, -2, 3, -4, 5]
>>>
Like Haskell's operator ":":
>>> from pipe import t
>>> for i in 0 | t(1) | t(2):
... print(i)
0
1
2
>>>
Yields the given quantity of the last elements of the given iterable.
>>> from pipe import tail
>>> for i in (1, 2, 3, 4, 5) | tail(3):
... print(i)
3
4
5
>>>
Yields the given quantity of elements from the given iterable, like head
in shell script.
>>> from pipe import take
>>> for i in count() | take(5):
... print(i)
0
1
2
3
4
>>>
Like itertools.takewhile
, yields elements of the
given iterable while the predicate is true:
>>> from pipe import take_while
>>> for i in count() | take_while(lambda x: x ** 2 < 100):
... print(i)
0
1
2
3
4
5
6
7
8
9
>>>
tee outputs to the standard output and yield unchanged items, useful for debugging a pipe stage by stage:
>>> from pipe import tee
>>> sum(["1", "2", "3", "4", "5"] | tee | map(int) | tee)
'1'
1
'2'
2
'3'
3
'4'
4
'5'
5
15
>>>
The 15
at the end is the sum
returning.
Transposes the rows and columns of a matrix.
>>> from pipe import transpose
>>> [[1, 2, 3], [4, 5, 6], [7, 8, 9]] | transpose
[(1, 4, 7), (2, 5, 8), (3, 6, 9)]
>>>
Recursively unfold iterables:
>>> list([[1, 2], [[[3], [[4]]], [5]]] | traverse)
[1, 2, 3, 4, 5]
>>> squares = (i * i for i in range(3))
>>> list([[0, 1, 2], squares] | traverse)
[0, 1, 2, 0, 1, 4]
>>>
Like dedup() but only deduplicate consecutive values, using the given
key
function if provided (or else the identity).
>>> from pipe import uniq
>>> list([1, 1, 2, 2, 3, 3, 1, 2, 3] | uniq)
[1, 2, 3, 1, 2, 3]
>>> list([1, -1, 1, 2, -2, 2, 3, 3, 1, 2, 3] | uniq(key=abs))
[1, 2, 3, 1, 2, 3]
>>>
Only yields the matching items of the given iterable:
>>> list([1, 2, 3] | where(lambda x: x % 2 == 0))
[2]
>>>
Don't forget they can be aliased:
>>> positive = where(lambda x: x > 0)
>>> negative = where(lambda x: x < 0)
>>> sum([-10, -5, 0, 5, 10] | positive)
15
>>> sum([-10, -5, 0, 5, 10] | negative)
-15
>>>
You can construct your pipes using the Pipe
class like:
from pipe import Pipe
square = Pipe(lambda iterable: (x ** 2 for x in iterable))
map = Pipe(lambda iterable, fct: builtins.map(fct, iterable)
>>>
As you can see it's often very short to write, and with a bit of luck the function you're wrapping already takes an iterable as the first argument, making the wrapping straight forward:
>>> from collections import deque
>>> from pipe import Pipe
>>> end = Pipe(deque)
>>>
and that's it itrable | end(3)
is deque(iterable, 3)
:
>>> list(range(100) | end(3))
[97, 98, 99]
>>>
In case it gets more complicated one can use Pipe
as a decorator to
a function taking an iterable as the first argument, and any other
optional arguments after:
>>> from statistics import mean
>>> @Pipe
... def running_average(iterable, width):
... items = deque(maxlen=width)
... for item in iterable:
... items.append(item)
... yield mean(items)
>>> list(range(20) | running_average(width=2))
[0, 0.5, 1.5, 2.5, 3.5, 4.5, 5.5, 6.5, 7.5, 8.5, 9.5, 10.5, 11.5, 12.5, 13.5, 14.5, 15.5, 16.5, 17.5, 18.5]
>>> list(range(20) | running_average(width=10))
[0, 0.5, 1, 1.5, 2, 2.5, 3, 3.5, 4, 4.5, 5.5, 6.5, 7.5, 8.5, 9.5, 10.5, 11.5, 12.5, 13.5, 14.5]
>>>
Sometimes you just want a one-liner, when creating a pipe you can specify the function's positional and named arguments directly
>>> from itertools import combinations
>>> list(range(5) | Pipe(combinations, 2))
[(0, 1), (0, 2), (0, 3), (0, 4), (1, 2), (1, 3), (1, 4), (2, 3), (2, 4), (3, 4)]
>>>
a simple running sum with initial starting value
>>> from itertools import accumulate
>>> list(range(10) | Pipe(accumulate, initial=1))
[1, 1, 2, 4, 7, 11, 16, 22, 29, 37, 46]
>>>
or filter your data based on some criteria
>>> from itertools import compress
list(range(20) | Pipe(compress, selectors=[1, 0] * 10))
[0, 2, 4, 6, 8, 10, 12, 14, 16, 18]
>>> list(range(20) | Pipe(compress, selectors=[0, 1] * 10))
[1, 3, 5, 7, 9, 11, 13, 15, 17, 19]
>>>
Find the sum of all the multiples of 3 or 5 below 1000.
>>> sum(count() | where(lambda x: x % 3 == 0 or x % 5 == 0) | take_while(lambda x: x < 1000))
233168
>>>
Find the sum of all the even-valued terms in Fibonacci which do not exceed four million.
sum(fib() | where(lambda x: x % 2 == 0) | take_while(lambda x: x < 4000000))
Find the difference between the sum of the squares of the first one hundred natural numbers and the square of the sum.
>>> square = map(lambda x: x ** 2)
>>> sum(range(101)) ** 2 - sum(range(101) | square)
25164150
>>>
A pipe
can be parametrized without being evaluated:
>>> running_average_of_two = running_average(2)
>>> list(range(20) | running_average_of_two)
[0, 0.5, 1.5, 2.5, 3.5, 4.5, 5.5, 6.5, 7.5, 8.5, 9.5, 10.5, 11.5, 12.5, 13.5, 14.5, 15.5, 16.5, 17.5, 18.5]
>>>
For multi-argument pipes then can be partially initialized, you can think of curying:
some_iterable | some_pipe(1, 2, 3)
some_iterable | Pipe(some_func, 1, 2, 3)
is strictly equivalent to:
some_iterable | some_pipe(1)(2)(3)
So it can be used to specialize pipes, first a dummy example:
>>> @Pipe
... def addmul(iterable, to_add, to_mul):
... """Computes (x to_add) * to_mul to every items of the input."""
... for i in iterable:
... yield (i to_add) * to_mul
>>> mul = addmul(0) # This partially initialize addmul with to_add=0
>>> list(range(10) | mul(10))
[0, 10, 20, 30, 40, 50, 60, 70, 80, 90]
Which also works with keyword arguments:
>>> add = addmul(to_mul=1) # This partially initialize addmul with `to_mul=1`
>>> list(range(10) | add(10))
[10, 11, 12, 13, 14, 15, 16, 17, 18, 19]
>>>
But now for something interesting:
>>> import re
>>> @Pipe
... def grep(iterable, pattern, flags=0):
... for line in iterable:
... if re.match(pattern, line, flags=flags):
... yield line
...
>>> lines = ["Hello", "hello", "World", "world"]
>>> for line in lines | grep("H"):
... print(line)
Hello
>>>
Now let's reuse it in two ways, first with a pattern:
>>> lowercase_only = grep("[a-z] $")
>>> for line in lines | lowercase_only:
... print(line)
hello
world
>>>
Or now with a flag:
>>> igrep = grep(flags=re.IGNORECASE)
>>> for line in lines | igrep("hello"):
... print(line)
...
Hello
hello
>>>
Pipe uses generators all the way down, so it is naturally lazy.
In the following examples we'll use itertools.count: an infinite generator of integers.
We'll make use of the tee
pipe too, which prints every values that
passe through it.
The following example does nothing, nothing is printed by tee
so no
value passed through it. It's nice because generating an infinite
sequence of squares is "slow".
>>> result = count() | tee | select(lambda x: x ** 2)
>>>
Chaining more pipes still won't make previous ones start generating
values, in the following example not a single value is pulled out of
count
:
>>> result = count() | tee | select(lambda x: x ** 2)
>>> first_results = result | take(10)
>>> only_odd_ones = first_results | where(lambda x: x % 2)
>>>
Same without variables:
>>> result = (count() | tee
... | select(lambda x: x ** 2)
... | take(10)
... | where(lambda x: x % 2))
>>>
Only when values are actually needed, the generators starts to work.
In the following example only two values will be extracted out of count
:
0
which is squared (to0
), passes thetake(10)
eaily, but is dropped bywhere
1
which is squared (to1
), also easily passes thetake(10)
, passes thewhere
, and passes thetake(1)
.
At this point take(1)
is satisfied so no other computations need to
be done. Notice tee
printing 0
and 1
passing through it:
>>> result = (count() | tee
... | select(lambda x: x ** 2)
... | take(10)
... | where(lambda x: x % 2))
>>> print(list(result | take(1)))
0
1
[1]
>>>
In pipe 1.x a lot of functions were returning iterables and a lot other functions were returning non-iterables, causing confusion. The one returning non-iterables could only be used as the last function of a pipe expression, so they are in fact useless:
range(100) | where(lambda x: x % 2 == 0) | add
can be rewritten with no less readability as:
sum(range(100) | where(lambda x: x % 2 == 0))
so all pipes returning non-iterables were deprecated (raising warnings), and finally removed in pipe 2.0.
Oh, you just upgraded pipe, got an exception, and landed here? You have three solutions:
-
Stop using closing-pipes, replace
...|...|...|...|as_list
tolist(...|...|...|)
, that's it, it's even shorter. -
If "closing pipes" are not an issue for you, and you really like them, just reimplement the few you really need, it often take a very few lines of code, or copy them from here.
-
If you still rely on a lot of them and are in a hurry, just
pip install pipe<2
.
And start testing your project using the Python Development Mode so you catch those warnings before they bite you.
This has already been discussed in #74.
An @Pipe
is often easily implemented in a 1 to 3 lines of code
function, and the pipe
module does not aim at giving all
possibilities, it aims at giving the Pipe
decorator.
So if you need more pipes, closing pipes, weird pipes, you-name-it, feel free to implement them on your project, and consider the already-implemented ones as examples on how to do it.
See the Constructing your own
paragraph below.