基于公共id的元组排序列表合并

2024-10-04 07:33:31 发布

您现在位置:Python中文网/ 问答频道 /正文

我有以下已排序的元组列表:

list1 = [(0.2, 'a'), (0.4, 'b'), (0.5,'d')]
list2 = [(0.1, 'a'), (0.3, 'c'), (0.7, 'x')]
list3 = [(0.5, 'c'), (0.6, 'a'), (0.5, 'b')]

我想根据常见字母创建一个总体排名列表,如下所示:

  1. 如果字母在所有三个列表中都是通用的,则添加三个单独的值
  2. 如果字母仅在两个列表之间通用,请将两个单独的值和1相加
  3. 如果元素仅在一个列表中,则在其值上添加2

预期结果:

[(0.9, 'a'), (1.8, 'c'), (1.9, 'b'), (2.5, 'd'), (2.7, 'x')]

工作原理:

如果该项在所有三个列表中都很常见,我就能够得到预期的结果,但是如果是其他情况,我就无法得到正确的结果

代码片段

list1 = [(0.2, 'a'), (0.4, 'b'), (0.5, 'd')]
list2 = [(0.1, 'a'), (0.3, 'c'), (0.7, 'x')]
list3 = [(0.5, 'c'), (0.6, 'a'), (0.5, 'b')]
priority_result = [] # when element is common in all 3 lists
twos_array = [] #when element is common in only two lists

result = [(s1, l1 + l1) for (l1, s1), (l1, s2) in zip(list1, list2)]
print(result)
for (score, resultID) in list1:
    for (score1, resultID1) in list2:
        for (score2, resultID2) in list3:
            if(resultID == resultID1 or resultID == resultID2):                    
                result = [(score + score1 + score2, resultID)]
                priority_result.extend(result)
            elif(resultID == resultID1 and resultID != resultID2):
                result = [(score + score1 + 1, resultID)]
                twos_array.extend(result)

我如何在这方面工作以产生预期的结果


Tags: inl1列表for字母resultscorepriority
3条回答

您可以尝试将^{}^{}一起使用:

from itertools import groupby
from operator import itemgetter
x = list1 + list2 + list3
y = [l[1] for l in x]
print(sorted([((3 - y.count(key)) + sum(next(zip(*l))), key) for key, l in groupby(sorted(x, key=ig(1)), key=ig(1))], key=ig(0)))

[(0.9, 'a'), (1.8, 'c'), (1.9, 'b'), (2.5, 'd'), (2.7, 'x')]

此代码连接列表,并仅为键创建另一个列表,并按键分组,对值进行求和。此外,它还根据出现的次数添加值的预期增加

最后,它根据求和值和修改值进行排序

可以交换元组的顺序以创建映射:

d1 = dict(x[::-1] for x in list1)
d2 = dict(x[::-1] for x in list2)
d3 = dict(x[::-1] for x in list3)

现在可以对键进行并集,因为dict.keys返回类似set的对象:

keys = d1.keys() | d2.keys() | d3.keys()

其余的可以通过dict.get完成:

result = {k: d1.get(k, 1) + d2.get(k, 1) + d3.get(k, 1) for k in keys}

将其转换为排序列表非常简单:

sorted(x[::-1] for x in result.items())

假设您的列表现在位于元列表中:

lists = [list1, list2, list3]
keys = set().union(*lists)
dicts = [dict(x[::-1] for x in l) for l in lists]
result = {k: sum(d.get(k, 1) for d in dicts) for k in keys}
result = sorted(x[::-1] for x in result.items())

下面是一个稍微简单的解决方案:

mapping = dict.fromkeys(set().union(*lists), len(lists))
for v, k in itertools.chain.from_iterable(lists):
    mapping[k] += v - 1
result = sorted(x[::-1] for x in result.items())

您可以使用collections.Counter为您完成大部分数学运算:

c = Counter()
for lst in lists:
    c.update({k: v - 1 for v, k in lst})
result = [(v + len(lists), k) for k, v in c.items()]

正则collections.defaultdict的相同之处是:

d = defaultdict(int)
for v, k in itertools.chain.from_iterable(lists):
    d[k] += v - 1
result = [(v + len(lists), k) for k, v in d.items()]
list1 = [(0.2, 'a'), (0.4, 'b'), (0.5, 'd')]
list2 = [(0.1, 'a'), (0.3, 'c'), (0.7, 'x')]
list3 = [(0.5, 'c'), (0.6, 'a'), (0.5, 'b')]

d = {}
for t in list1 + list2 + list3:
    d.setdefault(t[1], []).append(t[0])
lst = [(sum(v, 3 - len(v)), k) for k, v in d.items()]
print(lst)  # [(0.9, 'a'), (1.9, 'b'), (2.5, 'd'), (1.8, 'c'), (2.7, 'x')]

相关问题 更多 >