[英]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 如果 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)}
然后您可以遍历两个字典的键并进行比较。 假设您只想对t1
和t2
存在的键/元组进行比较,您可以使用集合连接键:
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
,您可以比较两个元组a
和b
列表,因此:
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.