Passing a parameter through a python filter function

Question

I'm trying to filter a list of words/phrases by some keyword. All the examples I've found of the filter() function have been using numbers, so I'm wondering if this is even possible. I know that filter() will put an item into the resulting list if the function it calls returns True.

Let's say I have something like this:

def filtCheck(item, filt):
    if filt in item:
        return True

def funct():
    filt = 'Hi'
    set1 = ['Hello, world', 'Hi there', 'Hi friend']
    set2 = filter(filtCheck(filt), set1)

    print set2

Here is where I get confused. How exactly would I write that first parameter on the set2 line? Obviously not the way its written since my filtCheck function takes two parameters and I only provide one. Do I need to modify my filtCheck function too? But if I take the item parameter out of it, there's no string to check if filt is in.

codeape · Accepted Answer · 2013-10-23 19:19:00Z

7

You could use a lambda (see also http://docs.python.org/2/reference/expressions.html#lambda):

set2 = filter(lambda item: filtCheck(item, 'Hi'), set1)

Or you could use functools.partial:

from functools import partial
set2 = filter(partial(filtCheck, filt="Hi"), set1)

Or you could skip the filter function and use a list comprehension:

set2 = [item for item in set1 if filtCheck(item, "Hi")]

# or simply

set2 = [item for item in set1 if "Hi" in item]

edited Oct 23, 2013 at 19:19

answered Oct 23, 2013 at 19:10

codeape

102k26 gold badges180 silver badges201 bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

Xonal Over a year ago

Went with your last option. Thanks a lot :)

Prashant Kumar · Accepted Answer · 2013-10-23 19:21:58Z

1

Easier to switch to a list-comp:

filt = 'Hi'
set1 = ['Hello, world', 'Hi there', 'Hi friend']
set2 = [greeting for greeting in set1 if filt in greeting]

edited Oct 23, 2013 at 19:21

Prashant Kumar

23.2k14 gold badges52 silver badges64 bronze badges

answered Oct 23, 2013 at 19:13

Jon Clements

143k34 gold badges254 silver badges288 bronze badges

Comments

A.E. Drew · Accepted Answer · 2013-10-23 19:23:13Z

1

You could use currying:

def filtCheck(filt):
    return lambda item: filt in item

def funct():
    filt = 'Hi'
    set1 = ['Hello, world', 'Hi there', 'Hi friend']
    set2 = filter(filtCheck(filt), set1)
    print set2

answered Oct 23, 2013 at 19:23

A.E. Drew

2,1571 gold badge16 silver badges26 bronze badges

Comments

John Spong · Accepted Answer · 2013-10-23 19:32:14Z

The most direct modification to this code would be to have filtCheck return a function rather than a boolean value:

def filtCheck(value):
    def is_in_list(lst):
        return value in lst
    return is_in_list

For your purposes, the function argument for filter is any function that takes exactly one parameter and returns a boolean value indicating whether or not it should be included in the result.

However, from the filter docs (http://docs.python.org/2/library/functions.html#filter),

Note that filter(function, iterable) is equivalent to [item for item in iterable if function(item)] if function is not None and [item for item in iterable if item] if function is None.

Combining this equivalency with the fact that filter is deprecated in python3, I would suggest you use list comprehensions:

def filtCheck(item, filt):
    if filt in item:
        return True

set2 = [item for item in set1 if filtCheck(filt, item)]

This can be further simplified to get rid of the need to use filtCheck altogether:

def funct():
    filt = 'Hi'
    set1 = ['Hello, world', 'Hi there', 'Hi friend']
    set2 = [item for item in set1 if filt in item]

    print set2

Roman · Accepted Answer · 2022-01-10 04:39:35Z

0

Keeping the construction of the example and taking a safe way, we need to pass an additional parameter with a default value:

def funct():
  filt = 'Hi'

  def filtCheck(item, filt=filt):
    if filt in item:
      return True

  set1 = ['Hello, world', 'Hi there', 'Hi friend']
  set2 = list(filter(filtCheck, set1))

  print(set2)

edited Jan 10, 2022 at 4:39

answered Jan 10, 2022 at 4:14

Roman

22.3k18 gold badges103 silver badges105 bronze badges

Collectives™ on Stack Overflow

Passing a parameter through a python filter function

5 Answers 5

1 Comment

Comments

Comments

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

5 Answers 5

1 Comment

Comments

Comments

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related