How to compare two lists to keep matching substrings?

Question

As best I can describe it, I have two lists of strings and I want to return all results from list A that contain any of the strings in list B. Here are details:

A = ['dataFile1999', 'dataFile2000', 'dataFile2001', 'dataFile2002']
B = ['2000', '2001']

How do I return

C = ['dataFile2000', 'dataFile2001']?

I've been looking into list comprehensions, doing something like below

C=[x for x in A if B in A] 

but I can't seem to make it work. Am I on the right track?

Answer 1

You were close, use any:

C=[x for x in A if any(b in x for b in B)]

More detailed:

A = ['dataFile1999', 'dataFile2000', 'dataFile2001', 'dataFile2002']
B = ['2000', '2001']

C = [x for x in A if any(b in x for b in B)]

print(C)

Output

['dataFile2000', 'dataFile2001']