在 Python 中与列表相交的最有效方法是什么？

Question

我之前问过一个关于阅读 CSV 和使用 Pandas 的问题，虽然这让我更接近答案，但这并不是我想要的。

我有两个列表：

a1 = [['1', 'foo'], ['2', 'bar'], ['3', 'Foo']]
a2 = [['1', 'foo'], ['2', 'bar'], ['3', 'Foo'], ['4', 'Bar']]

我正在寻找这些东西：

• a1 中不在 a2 中的所有项目，搜索第一个值
• a2 中不在 a1 中的所有项目，搜索第一个值
• a1 和 a2 中的所有项目，搜索第一个值
  • 对于此交集中的每个项目，第二个值匹配的所有项目的列表
  • 对于此交集中的每个项目，列出第二个值不匹配的所有项目

我从我想要使用 Pandas 的 CSV 文件中获取数据，现在我以上面显示的列表格式获取它们。提前致谢！

Answer 1

最简单的解决方案是将其转换为集合

a1 = [['1', 'foo'], ['2', 'bar'], ['3', 'Foo']]
a2 = [['1', 'foO'], ['2', 'bar'], ['3', 'Foo'], ['4', 'Bar']]

a1_set = set([v[0] for v in a1])
a2_set = set([v[0] for v in a2])

print("Items in a1,but not a2:",[x for x in a1 if x[0] in a1_set-a2_set])
print("Items in a2,but not a1:",[x for x in a2 if x[0] in a2_set-a1_set])

d1 = dict(a1)
d2 = dict(a2)
items_in_both = [(k,(d1[k],d2[k]) if d1[k] != d2[k] else d1[k]) for k in a2_set&a1_set]

print("Items in both sets:", items_in_both)

Answer 2

所以，基本上你想找到一个列表中而不是另一个列表中的元素：

[x for x in a2 if x not in a1]

如果你想要通用元素，你可以使用：

[x for x in a2 if x in a1]