python - 如何在两个具有相同元素的 Python 列表上执行 MYSQL INNER JOIN？

Question

我有两个 python3 数组(列表)，类似于 MYSQL 中的两列。他们的一些第一个“列”元素是相同的:

list1 = [ 

['a', 'apple'], 
['b', 'banana'], 
['m', 'mango']

]

list2 = [ 

['b', 'buffalo'], 
['zzzz', 'zzzzegemot, a sleeping behemoth'], 
['m', 'mongoose']

]

请注意，在两个列表中，第一个“b”和第三个“m”元素是相同的。

如何在 list1[0] 和 list2[0] 上执行类似于 MYSQL INNER JOIN 的操作，并获得第三个列中类似条目的组合:

list3 = [

['banana', 'buffalo']
['mango', 'mongoose']

]

我希望我表达清楚，以便您理解我。

Answer 1

由于您有两个列表，内部连接应该处理重复的键。

m1 = reduce(lambda c,x: c[x[0]].append(x[1]) or c,list1,defaultdict(list))
list(chain(*(product(m1[e[0]],[e[1]]) for e in list2)))

# added entries ['b', 'bongo']  and ['b', 'bull'] into list1 and list2
>>> [('banana', 'buffalo'),
     ('bongo', 'buffalo'),
     ('banana', 'bull'),
     ('bongo', 'bull'),
     ('mango', 'mongoose')]