Check to see if a string contains either of the strings in a list in Python

Question

I have a list a:

a = ['2005', '2006']

and list b:

b = ['2005.123', '2005.234', '2006.123', '2007.234']

If I wanted a list of all the elements in b that contains a string '2005', I would do:

[value for value in b if '2005' in str(value)]

But I would like a list of all the values in b that contain either of the string values in list a (list a and b can be several elements long), and this one does not work:

[value for value in b if ['2005', '2006'] in str(value)]

I would like the code to return

['2005.123', '2005.234', '2006.123']

Is there an easier way to do this, without using loops?

pault · Accepted Answer · 2018-03-08 22:26:59Z

4

Use any():

[value for value in b if any(d in value for d in a)]
#['2005.123', '2005.234', '2006.123']

Also, you do not have to call str(value) as it's already a string.

Update: Just for fun, here's another way to do the same thing using filter():

filter(lambda value: any(d in value for d in a), b)

edited Mar 8, 2018 at 22:26

answered Mar 8, 2018 at 22:18

pault

43.7k17 gold badges121 silver badges161 bronze badges

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

1 Comment

DPdl Over a year ago

Perfect! Thank you. The reason for str(value) was to allow for the chance that some numbers may not have been represented as string.

Steven Rumbalski · Accepted Answer · 2018-03-08 22:32:36Z

2

You ask for "contains" but your example has prefixes. If it's prefixes you want use str.startswith, which will allow you to pass a tuple of prefixes as its first argument:

a_tup = tuple(a)
[value for value in b if value.startswith(a_tup)]

answered Mar 8, 2018 at 22:32

Steven Rumbalski

45.7k10 gold badges96 silver badges125 bronze badges

Collectives™ on Stack Overflow

Check to see if a string contains either of the strings in a list in Python

2 Answers 2

1 Comment

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

1 Comment

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related