python - 将列表与列表列表进行比较，并从列表列表中返回另一个元素

Question

我有一个 list - 答案:

[0,1] 

和另一个列表列表 - 问题:

[[0,ABC], [1,DEF], [3,XYZ]]

我如何比较 2 并返回

 ABC, DEF 

基于答案中的所有元素与列表列表中第一个元素的比较？

Answer 1

您可以使用 list comprehension并将第二项保留在 questions 中 如果第一项包含在 answers 中:

answers = set([0,1])
[i[1] for i in questions if i[0] in answers]
# ['ABC', 'DEF']

---

注意:使用集合而不是列表有助于将复杂度从 O(N*M) 降低到 O(N)，因为@RafaelC 建议，N 是问题的长度，M 是答案的长度