Проверка наличия строки в списке в другом списке строк Python

Question

Я пытаюсь проверить, есть ли строка в test_list2 в каждой строке test_list1. Если строка в test_list2 присутствует в test_list1, я хочу добавить 1 в новый список, иначе добавьте 0 в тот же новый список.

Итак:

test_list1 = [['Happy', 'Dog'], ['Sad', 'Dog'], ['Dog', 'Dog'], ['Angry', 'Dog']] 
test_list2 = ['Happy', 'Sad', 'Angry'] 
new_list = [] 

def emotion_detection(x, y): 
    for i in x: 
     if i in y: 
      new_list.append(1) 
     if i not in y: 
      new_list.append(0) 
    print new_list 

возвращается: [0, 0, 0, 0], когда я думаю, что он должен вернуться [1,1,0,1]

Заранее спасибо за любую помощь !

Answer 1

Это логическое поведение, которое вы ожидаете?

test_list1 = [['Happy', 'Dog'], ['Sad', 'Dog'], ['Dog', 'Dog'], ['Angry', 'Dog']] 
test_list2 = ["Happy", "Sad", "Angry"] 
new_list = [] 

def emotion_detection(x, y): 
    for i in x: 
     for string in i: 
      if string in y: 
       new_list.append(1) 
       break 
     else: 
      new_list.append(0) 
    print new_list 

Answer 2

test_list1 = [['Happy', 'Dog'], ['Sad', 'Dog'], ['Dog', 'Dog'], ['Angry', 'Dog']] 
test_list2 = ["Happy", "Sad", "Angry"] 
new_list = [] 

for element in test_list1: 
    if element[0] in test_list2: 
     new_list.append(1) 
    else: 
     new_list.append(0) 
print new_list 

вместо использования i используйте i[0], который является индексом эмоций!

Answer 3

emotions = set(["Happy", "Sad", "Angry"]) 
sentences = ['Happy Dog', 'Sad Dog', 'Dog Dog', 'Angry Dog'] 

def is_emotional(sentence): 
    words = sentence.split() 
    return bool(emotions.intersection(words)) 

sentence_emotions = [is_emotional(sentence) for sentence in sentences] 
# gives [True, True, False, True] 

Edit: Hackaholic имеет хорошую идею с any, но я бы написать это как

def is_emotional(sentence): 
    return any(word in emotions for word in sentence.split()) 

, который может быть быстрее, особенно для длинных предложений, так как он останавливается, как только как слово эмоции найдено.

Answer 4

попробовать так:

>>> test_list1 = [['Happy', 'Dog'], ['Sad', 'Dog'], ['Dog', 'Dog'],['Angry', 'Dog']] 
>>> test_list2 = ["Happy", "Sad", "Angry"] 
>>> [any(set(x)&set(test_list2)) for x in test_list1] 
[True, True, False, True] 

Answer 5

Использование списковых:

new_list = [k[0] in test_list2 for k in test_list1]