python - 根据元素值从嵌套列表中删除子列表

Question

我正在尝试根据元素值从嵌套列表中删除子列表。

data = [['1', 'i like you'],
['2', 'you are bad'],
['5', 'she is good'],
['7', 'he is poor']]

negative_words = set(['poor', 'bad'])

如果第二列包含否定词，我想删除子列表。因此，期望的结果如下。有什么建议吗？

data = [['1', 'i like you'],
['5', 'she is good']]

Answer 1

我很容易想到两种方法。

>>> data = [['1', 'i like you'], ['2', 'you are bad'],
            ['5', 'she is good'], ['7', 'he is poor']]
>>> neg_words = {'poor', 'bad'}

使用将子列表中的字符串转换为集合，然后检查是否为 disjoint与neg_words，像这样

>>> [[n, ws] for n, ws in data if set(ws.split()).isdisjoint(neg_words)]
[['1', 'i like you'], ['5', 'she is good']]

或者简单地检查列表中字符串中的单词是否不在 neg_words 中，如下所示

>>> [[n, ws] for n, ws in data if all(w not in neg_words for w in ws.split())]
[['1', 'i like you'], ['5', 'she is good']]