python if/else 语句，如果存在于另一个子列表中，则保留元素

Question

假设我在 list 中有以下标记：

['I', 'want', 'to', 'learn', 'coding', 'in', 'r', 'and', 'c++', 'today', ',', 'and', 'then', 'I', "'ll", 'be', 'learning', 'c#', 'and', 'c']

我想删除所有长度 < 2 的标记，但想保留此子列表的元素：

['r','c','js','c#']

如何在单个 python list-comprehension 中做到这一点？

Answer 1

您可以在两个条件下使用list comprehension 。

lst1 = ['I', 'want', 'to', 'learn', 'coding', 'in', 'r', 'and', 'c++', 'today', ',', 'and', 'then', 'I', "'ll", 'be', 'learning', 'c#', 'and', 'c']

lst2 = ['r','c','js','c#']

res = [l for l in lst1 if (len(l)>=2) or (l in lst2)]
print(res)

---

['want',
 'to',
 'learn',
 'coding',
 'in',
 'r',
 'and',
 'c++',
 'today',
 'and',
 'then',
 "'ll",
 'be',
 'learning',
 'c#',
 'and',
 'c']