我想要获取两个列表并找出同时出现在这两个列表中的值。
a = [1, 2, 3, 4, 5] b = [9, 8, 7, 6, 5] returnMatches(a, b)
[5]例如将返回。
[5]
您可以使用 Python 的集合运算来实现这一点,它提供了一种简单而有效的方法来查找两个列表的交集。以下是一个接受两个列表并返回公共元素的函数:
Python复制代码def returnMatches(a, b): return list(set(a) & set(b)) # Example usage a = [1, 2, 3, 4, 5] b = [9, 8, 7, 6, 5] print(returnMatches(a, b)) # Output: [5]
set(a)
a
set(b)
b
set(a) & set(b)
list(set(a) & set(b))
如果你不想使用集合,你可以使用列表推导来实现相同的结果:
def returnMatches(a, b): return [item for item in a if item in b] # Example usage a = [1, 2, 3, 4, 5] b = [9, 8, 7, 6, 5] print(returnMatches(a, b)) # Output: [5]
此方法遍历列表中的每个元素a,如果它也存在于列表中,则将其包含在结果中b。
in
选择最适合您的需求和编码风格的方法。