比较元组元素列表 python

comparing list of tuple elements python

我有两个元组列表

t1 = [ ('a',3,4), ('b',3,4), ('c',4,5) ]
t2 = [ ('a',4,6), ('c',3,4), ('b',3,6), ('d',4,5) ]

这样

  1. 元组的顺序可能与
  2. 不同
  3. 列表可能包含不同数量的元组元素。

我的目标是比较两个列表,如果字符串元素匹配,则比较元组中的最后一个整数元素和 return 如果 t1[2] < t2[2,则包含 -1 的列表], 如果它们相等则为 0,如果它们大于则为 1。

我尝试了不同的变体,但我遇到的问题是找到一种方法来匹配字符串以进行正确的比较。

return [diff_unique(x[2],y[2]) for x,y in zip(new_list,old_list) ]

其中diff_unique做前面提到的整数比较,new_list是t1,old_list是t2。

我也试过这个:

return [diff_unique(x[2],y[2]) for x,y in zip(new_list,old_list) if(x[0]==y[0]]

我打算做的是使用 returned 列表并创建一个新的四元组列表,其中包含原始 t1 值以及与匹配的 t2 元组的差异。即

inc_dec_list = compare_list(new,old)
final_list = [ (f,r,u,chge) for (f,r,u), chge in zip(new,inc_dec_list)]

其中新 = t1,旧 = t2。这可能是一个重要的细节,抱歉我错过了。

有正确方向的帮助吗?

编辑:我已经添加了我的测试用例程序,它模仿了我的初衷是为了那些想要帮助的人。谢谢大家

import os 
import sys

old = [('a',10,1),('b',10,2),('c',100,4),('d',200,4),('f',45,2)]
new = [('a',10,2),('c',10,2),('b',100,2),('d',200,6),('e',233,4),('g',45,66)]


def diff_unique(a,b):
    print "a:{} = b:{}".format(a,b)
    if a < b:
        return -1
    elif a==b:
        return 0
    else:
        return 1

def compare_list(new_list, old_list):
    a = { t[0]:t[1:] for t in new_list } 
    b = { t[0]:t[1:] for t in old_list }
    common = list( set(a.keys())&set(b.keys()))
    return [diff_unique(a[key][1], b[key][1]) for key in common]
    #get common tuples
    #common = [x for x,y in zip(new_list,old_list) if x[0] == y[0] ]
    #compare common to old list
    #return [diff_unique(x[2],y[2]) for x,y in zip(new_list,old_list) ]

inc_dec_list = compare_list(new,old)
print inc_dec_list

final_list = [ (f,r,u,chge) for (f,r,u), chge in zip(new,inc_dec_list)]
print final_list

首先,从每个列表构建一个 default dictionary,不存在的键的默认值是一个元组,其最后一个元素是比较的最小可能值。

SMALL = (-float['inf'],)

from collections import defaultdict
d1 = defaultdict(lambda: SMALL, [(t[0], t[1:]) for t in t1])
d2 = defaultdict(lambda: SMALL, [(t[0], t[1:]) for t in t2])

接下来,迭代每个字典中的键(可以使用 itertools.chain 轻松创建)。您可能希望对结果列表的键进行排序以使其具有任何意义(否则,您如何知道哪些键产生了 -1/0/1 中的哪一个?)

from itertools import chain
all_keys = set(chain(d1, d2))
result = [cmp(d1[k][-1], d2[k][-1]) for k in sorted(all_keys)]

这是您问题的简单解决方案, 这不是您尝试过的一行。我希望它仍然对你有帮助

for a in t1:
    for b in t2:
        if a[0] != b[0]:
            continue
        return cmp(a[-1], b[-1])

要通过来自不同列表的字符串匹配元组,您可以使用字典理解(保留元组内的顺序):

a = {t[0]:t[1:] for t in t1} # {'a': (3, 4), 'c': (4, 5), 'b': (3, 4)}
b = {t[0]:t[1:] for t in t1} # {'a': (4, 6), 'c': (3, 4), 'b': (3, 6), 'd': (4, 5)}

然后你可以遍历两个字典的键并进行比较。假设您只想比较 t1 t2 中的 keys/tuples,您可以使用集合连接键:

common_keys = list(set(a.keys())&set(b.keys()))

最后比较字典的项目并像这样创建您想要的列表:

return [diff_unique(a[key][1],b[key][1]) for key in common_keys ]

如果您需要按字母顺序排列字符的顺序输出,请在键上使用 sorted 函数:

return [diff_unique(a[key][1],b[key][1]) for key in sorted(common_keys) ]

如果要考虑所有键,可以执行以下操作:

all_keys = list(set(a.keys()+b.keys()))
l = list()
for key in sorted(all_keys):
    try:
        l.append(diff_unique(a[key][1],b[key][1]))
    except KeyError:
        l.append("whatever you want")
return l

有了关于应该以什么顺序返回什么值的新信息,解决方案是这样的:

ordered_keys = [t[0] for t in t1]
a = {t[0]:t[1:] for t in t1} # {'a': (3, 4), 'c': (4, 5), 'b': (3, 4)}
b = {t[0]:t[1:] for t in t1} # {'a': (4, 6), 'c': (3, 4), 'b': (3, 6), 'd': (4, 5)}
l = list()
for key in sorted(ordered_keys):
    try:
        l.append(diff_unique(a[key][1],b[key][1]))
    except KeyError:
        l.append(0) # default value
return l

python 3.x中,你可以比较两个元组列表 ab 因此:

import operator

a = [(1,2),(3,4)]
b = [(3,4),(1,2)]
# convert both lists to sets before calling the eq function
print(operator.eq(set(a),set(b))) #True