在字符串列表中查找特定的单词组

Question

Let's say I have a list of strings, generated from a text:

string_lst = ['A', 'cat', 'is', 'a', 'cat', 'but', 'a', 'dog', 'is', 'not', 'a', 'cat']

( The condition is to keep the string list separated as is. )

Imagine that the strings have the following assignments:

A | WORD

cat | ANIMAL

is | WORD

a | WORD

cat | ANIMAL

but | WORD

a | WORD

dog | ANIMAL

is | WORD

not | WORD

a | WORD

cat | ANIMAL

I need you to find only the strings, in their natural sequence, ( ANIMAL + WORD + WORD + ANIMAL ), that is, give me a group of words in that exact sequence. Using the example above, the result will be: 'cat is a cat', 'cat but a dog'

Any idea?

Answer 1

You just need a list of all animals

string_lst = ['A', 'cat', 'is', 'a', 'cat', 'but', 'a', 'dog', 'is', 'not', 'a', 'cat']

animals = ['dog', 'cat']
result = []
for i in range(len(string_lst) - 4):
    if string_lst[i] in animals and string_lst[i + 1] not in animals and string_lst[i + 2] not in animals and string_lst[i + 3] in animals:
        result.append(string_lst[i:i + 4])

print(result)