python如何求嵌套列表的交集？

[ImN1]

a = [[8, 6, 3], [3, 4, 5], [5, 2, 7]]
b = [[2, 5, 4], [6, 8, 5]]

python3.3
求a[i][0]==b[j][1] and a[i][1]==b[j][0]同时满足的情况

返回 [[8, 6, 3], [5, 2, 7]]

a,b各嵌套一层，但可能有数万个item，以减少循环为佳，故不想用穷举方法

[qq120848369]

这么说就很简单了, 用每一个元素的前2个元素作为key(b的要把顺序颠倒一下), a和b分别做自己的映射, 最后直接就得到了.

#python 3.2
#coding=gb2312

def make_mapping(data, reverse = False):
    mapping = {}
    for elem in data:
        if reverse:
            key = str(elem[1]) + ":" + str(elem[0])
        else:
            key = str(elem[0]) + ":" + str(elem[1])
        if key in mapping:
            mapping[key].append(elem)
        else:
            mapping[key] = [elem]
    return mapping

def make_result(mapping_a, mapping_b):
    result = []
    for key, lst in mapping_a.items():
        if key in mapping_b:
            result.extend(lst)
    return result

if __name__ == "__main__":
    a = [[8, 6, 3], [3, 4, 5], [5, 2, 7]]
    b = [[2, 5, 4], [6, 8, 5]]
    mapping_a = make_mapping(a)
    mapping_b = make_mapping(b, True)
    result = make_result(mapping_a, mapping_b)
    print(result)

[[8, 6, 3], [5, 2, 7]]

[ImN1]

映射还不会，先留下做例程学习，嘿嘿

[ImN1]

引用 1 楼 qq120848369 的回复:

这么说就很简单了, 用每一个元素的前2个元素作为key(b的要把顺序颠倒一下), a和b分别做自己的映射, 最后直接就得到了.

初步看了一看，用其中的值组成字典吧？这个之前也想过，因为写惯了php的数组，和py的字典很相似 但担心是否会范围py的key规定，因为值不仅数字，有可能是unicode字串，用韩文测试了下也可以……

[angel_su]

tuple元组也可以做key，按你的需求转化b成集合或字典即可... >>> a = [[8, 6, 3], [3, 4, 5], [5, 2, 7]] >>> b = [[2, 5, 4], [6, 8, 5]] >>> set_b = {(x, y) for x, y, z in b} >>> set_b set([(2, 5), (6, 8)]) >>> [i for i in a if (i[1], i[0]) in set_b] [[8, 6, 3], [5, 2, 7]] >>>

[ImN1]

这个简单点，偶喜欢……也看得明白，嘿嘿

引用 4 楼 angel_su 的回复:

tuple元组也可以做key，按你的需求转化b成集合或字典即可... >>> a = [[8, 6, 3], [3, 4, 5], [5, 2, 7]] >>> b = [[2, 5, 4], [6, 8, 5]] >>> set_b = {(x, y) for x, y, z in b} >>> set_b set([(2, 5), (6, 8)]) >……