繁体   English   中英

比较元组元素列表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. 列表可能不包含相同数量的元组元素。

我的目标是比较两个列表,如果字符串元素匹配,则比较元组中的最后一个整数元素并返回一个包含 -1 如果 t1[2] < t2[2] 的列表,如果它们相等则返回 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]]

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

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

要按不同列表中的字符串匹配元组,您可以使用 dict comprehension(保留元组内的顺序):

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)}

然后您可以遍历两个字典的键并进行比较。 假设您只想对t1t2存在的键/元组进行比较,您可以使用集合连接键:

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

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

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])

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

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM