如何比较 Pyhton 3.8 中的两个列表元素？

Question

例如：

如果两个列表元素是

a = [1,2,3,4,5]
b = [2,3,4,5,6]

我想得到[2,3,4,5]因为它共享相同的数字？ 有人可以帮助我吗？

哦，顺便问一下，如果 a 和 b 是随机列表，如何编写代码？

Answer 1

您可以使用列表推导或设置联合：

a = [1,2,3,4,5]
b = [2,3,4,5,6]

res = [x for x in a if x in b]
res_set = set(a) & set(b)

print(res)     # [2, 3, 4, 5]
print(res_set) # {2, 3, 4, 5}

Answer 2

a = [1,2,3,4,5]
b = [2,3,4,5,6]

c=[x for x in a if x in b]
d=[y for y in (a+b) if y not in c]

print(c)    # [2, 3, 4, 5]
print(d)    # [1,6]

您可以获取相同的元素并获取不同的元素。