CODEWITHSUNDEEP
pythonlistfilterfunctional-programming
kscottz
kscottz

Reputation: 1082

Python list filtering with arguments

Is there a way in python to call filter on a list where the filtering function has a number of arguments bound during the call. For example is there a way to do something like this:

>> def foo(a,b,c):
    return a < b and b < c

>> myList = (1,2,3,4,5,6)

>> filter(foo(a=1,c=4),myList)
>> (2,3)

This is to say is there a way to call foo such that a=1, c=4, and b gets bound to the values in myList?

Upvotes: 34

Views: 34951

Answers (3)

Zippo Lee
Zippo Lee

Reputation: 31

def foo(a,c):
    return lambda b : a < b and b < c

myList = (1,2,3,4,5,6)

g = filter(foo(1,4),myList)

Upvotes: 1

Zaur Nasibov
Zaur Nasibov

Reputation: 22659

You can create a closure for this purpose:

def makefilter(a, c):
   def myfilter(x):
       return a < x < c
   return myfilter

filter14 = makefilter(1, 4)

myList = [1, 2, 3, 4, 5, 6]
filter(filter14, myList)
>>> [2, 3]

Upvotes: 59

senderle
senderle

Reputation: 151007

One approach is to use lambda:

>>> def foo(a, b, c):
...     return a < b and b < c
... 
>>> myTuple = (1, 2, 3, 4, 5, 6)
>>> filter(lambda x: foo(1, x, 4), myTuple)
(2, 3)

Another is to use partial:

>>> from functools import partial
>>> filter(partial(foo, 1, c=4), myTuple)
(2, 3)

Upvotes: 70

Related Questions