python 列表取交集

Question

{代码...} 我想取 a 和 b 的交集，结果应该是 [3,4],[3,5] ,就是第一元素要一样我的笨办法是先取 a 和 b 第一个元素的列表，取它们的交集，然后再分别从 a 和 b 中找请问有没有更好的办法？ 补充：对不起大家我没...

怪我咯 · Answer

因为还不知道你想要的细节, 就先给一个暂时的答案吧:

from collections import defaultdict

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

dic_a = defaultdict(list)
dic_b = defaultdict(list)

for item in a:
    dic_a[item[0]].append(item)

for item in b:
    dic_b[item[0]].append(item)

common_keys = set(dic_a.keys()) & set(dic_b.keys())

for key in common_keys:
    print 'key:', key, 'a:', dic_a[key], 'b:',dic_b[key]

我回答过的问题: Python-QA

天蓬老师 · Answer

我的思路就是建立字典，然后直接查找

a = [[1,2],[3,4]]
b = [[2,3],[3,5]]
dic_a = {}
dic_b = {}
for item in a:
    if item[0] in dic_a:
        dic_a[item[0]].append(item)
    else:
        dic_a[item[0]] = item
for item in b:
    if item[0] in dic_b:
        dic_b[item[0]].append(item)
    else:
        dic_b[item[0]] = item
for key in dic_a:
    if key in dic_b:
        print dic_a[key],dic_b[key]

PHP中文网 · Answer

不应该是这样吗?

list(set(a).intersection(set(b)))

修改:

这样行不行

a = [[1, 2], [3, 4], [2, 6]]
b = [[2, 3], [3, 5],[1,5]]
for i in itertools.product(a, b):
    if i[0][0] == i[1][0]:
        print(i)

伊谢尔伦 · Answer

我不是来回答问题的，我看的楼主在用line_profiler，我想问一下，我刚下载了line_profiler，不知道安装好了没，我下载的源码，然后直接运行python setup.py install 但是用的时候都会报错：NameError: name 'profile' is not defined.不知道是什么原因，应该怎么解决？