比较两个元组列表中元素之间的元素

时间:2012-10-31 22:16:50

标签: python

这是我要做的事情。我有两个元组列表。构建一个元素列表,使list1中元组中的第一个元素与列表2中元组中的第一个元素匹配

list1 = [('a', 2), ('b', 3), ('z', 5)]

list2 = [('a', 1), ('b', 2), ('c', 3)]

list3 = ['a','b']

注意:第一个元素没有重复

在查看python列表推导之后,这就是我所做的

[x[0] for x in list1 if (x[0] in [y[0] for y in list2])]

我的问题是,有经验的python程序员会如何编写代码?自己编写了这个,我仍然觉得这很难读。如果不是,你会怎么做呢

4 个答案:

答案 0 :(得分:8)

我会使用zip()

In [25]: l1 = [('a', 2), ('b', 3), ('z', 5)]

In [26]: l2 = [('a', 1), ('b', 2), ('c', 3)]

In [27]: [x[0] for x,y in zip(l1,l2) if x[0]==y[0]]
Out[27]: ['a', 'b']

编辑:在阅读完上述评论后,看起来您正在寻找类似的内容:

In [36]: [x[0] for x in l1 if any(x[0]==y[0] for y in l2)]
Out[36]: ['a', 'b']

或使用sets

In [43]: from operator import itemgetter

In [44]: set(map(itemgetter(0),l1)) & set(map(itemgetter(0),l2))
Out[44]: set(['a', 'b'])

答案 1 :(得分:3)

我想你想在这里使用set

set(x[0] for x in list1).intersection(y[0] for y in list2)

或使用语法糖:

{x[0] for x in list1} & {y[0] for y in list2}

两者都导致:

set(['a', 'b'])

答案 2 :(得分:2)

我认为在这里使用集合可能更清楚(因为你没有重复的元素):

set1 = set( el[0] for el in list1 )
set2 = set( el[0] for el in list2 )
set3 = set1 & set2 # set intersection
# list3 = list(set3)

答案 3 :(得分:1)

假设您想要元组的第一个元素的集合交集,您可以使用Python 2.7中引入的字典键视图:

dict(list1).viewkeys() & dict(list2).viewkeys()

这比长列表的解决方案更有效,因为它具有线性运行时(与解决方案的O(mn)相反),但以任意顺序返回结果(与{定义的顺序相反) {1}})。

在Python 3.x中,这将是

list1