How to check if element in array contains any values from a list Python

Question

I have a list of words:

list1 = ['foo', 'baz', 'bat']

And an array of strings:

list2 = ['foo word word word', 'word baz word word', 'word word word word', 'word word bat word']
string_arr = np.array(list2)

I want to enumerate through the array and return a boolean if the element in the array contains any of the values in the list. The output to be a list of booleans:

[True, True, False, True]

Right now all I have is this code which just gives me a list of indices, which I don't want:

idx_mask = [idx for idx, element in enumerate(string_arr) if any(x in element for x in list1)]

How can I just get a list of booleans?

Do you want the text to contain a substring or specifically a word? if list1 contains w, should result be True on all of them? — Mike B
– Mike B, Commented Mar 6, 2021 at 21:50
@MikeB No, only if it contains the full word, the answer below works — 5sWithCrackScreen
– 5sWithCrackScreen, Commented Mar 6, 2021 at 21:53
Making list2 an array does not gain you anything. It's faster to iterate through a list, and numpy does not implement any special, faster, string methods. — hpaulj
– hpaulj, Commented Mar 6, 2021 at 21:53
@5sWithCrackScreen - no it doesn't. It works only for this example. Add 'w' to the list1 and all answers will be True — Mike B
– Mike B, Commented Mar 6, 2021 at 22:09
@MikeB Yes, but I'm only looking for like words in my case. Basically the SQL equivalent of like '%foo% — 5sWithCrackScreen
– 5sWithCrackScreen, Commented Mar 6, 2021 at 22:13

Alex Hall · Accepted Answer · 2021-03-06 21:47:20Z

2

print([any(x in element for x in list1) for element in list2])

answered Mar 6, 2021 at 21:47

Alex Hall

36.2k5 gold badges63 silver badges98 bronze badges

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

Comments

Mike B · Accepted Answer · 2021-03-06 22:11:57Z

1

To find only full word matches, you should match them to each string from list2 split by space to make a word array:

print([any(x in element.split(' ') for x in list1) for element in list2])

Test:

list1 = ['foo', 'baz', 'bat', 'w']
list2 = ['foo word word word', 'word baz word word', 'word word word word', 'word word bat word']

results are:
[True, True, False, True]
which is the expected result.

answered Mar 6, 2021 at 22:11

Mike B

2,7942 gold badges18 silver badges28 bronze badges

Collectives™ on Stack Overflow

How to check if element in array contains any values from a list Python

2 Answers 2

Comments

Comments

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Related