Question

这个问题是关于嵌套字典理解的问题，在提出这个问题之前我已经引用了link1和link2。
我有一个列表，其第一个元素是None，其余的是已排序的正数列表。

sorted_ar = [None, 10, 10, 12, 12, 12, 15, 25]

我的要求是建立一个字典：

key_dict = {10: [3, 2], 12: [12, 3], 15: [6, 1], 25: [7, 1]}

字典的值是两个元素列表，第一个元素是键的出现的索引的总和，第二个是出现的次数。
例如对于元素12，指数之和= 3 + 4 + 5 = 12，出现次数为3 以下代码执行此操作。

 key_dict = {k:[0,0] for k in sorted_ar if k!=None}
        for i in range(len(sorted_ar)):
            if sorted_ar[i]:
                key_dict[sorted_ar[i]][0] += i
                key_dict[sorted_ar[i]][1] += 1

我的要求是使用字典理解准备key_dict字典。

我的尝试：

key_dict = { 
    sorted_ar[i]:[ key_dict[sorted_ar[i]][0] + i,key_dict[sorted_ar[i]][0] + 1] 
    for i in range(1,len(sorted_ar)) if sorted_ar[i]!=None
}

但这会产生一些错误的结果

key_dict = {10: [2, 1], 12: [5, 1], 15: [6, 1], 25: [7, 1]}

在这种情况下我应该如何编写词典理解？

Answer 1

如果您打算使用排序，请查看itertools.groupby()和enumerate() function以添加索引：

from itertools import groupby

filtered = ((i, v) for i, v in enumerate(sorted_ar) if v)
grouped = ((v, list(g)) for v, g in groupby(filtered, lambda iv: iv[1]))
result = {v: [sum(i for i, v in g), len(g)] for v, g in grouped}

如果您愿意，可以将这些全部放入单个表达式中：

result = {v: [sum(i for i, v in g), len(g)] for v, g in (
    (v, list(g)) for v, g in groupby((
        (i, v) for i, v in enumerate(sorted_ar) if v), lambda iv: iv[1]))}

演示：

>>> from itertools import groupby
>>> sorted_ar = [None, 10, 10, 12, 12, 12, 15, 25]
>>> filtered = ((i, v) for i, v in enumerate(sorted_ar) if v)
>>> grouped = ((v, list(g)) for v, g in groupby(filtered, lambda iv: iv[1]))
>>> {v: [sum(i for i, v in g), len(g)] for v, g in grouped}
{10: [3, 2], 12: [12, 3], 15: [6, 1], 25: [7, 1]}

或作为一个长表达式：

>>> {v: [sum(i for i, v in g), len(g)] for v, g in ((v, list(g)) for v, g in groupby(((i, v) for i, v in enumerate(sorted_ar) if v), lambda iv: iv[1]))}
{10: [3, 2], 12: [12, 3], 15: [6, 1], 25: [7, 1]}

另一方面，您的字典方法不需要对输入进行排序，因此可以在O（N）时间内运行（排序需要O（NlogN）时间）。

Answer 2

你可以试试这个：

sorted_ar = [None, 10, 10, 12, 12, 12, 15, 25] 
new_data = {i:[sum(c for c, b in enumerate(sorted_ar) if b == i), sorted_ar.count(i)] for i in sorted_ar if i}

输出：

{25: [7, 1], 10: [3, 2], 12: [12, 3], 15: [6, 1]}

Answer 3

好的我找到了一种方法来实现我的行为，不知道为什么，但诀窍是字典更新需要在理解之外发生。

def my_summer(i,num,key_dict):
    key_dict[num][0] = key_dict[num][0]+i
    return key_dict[num][0]  


def my_counter(num,key_dict):
    key_dict[num][1] +=1
    return key_dict[num][1]

sorted_ar = [None, 10, 10, 12, 12, 12, 15, 25] 
key_dict = {k:[0,0] for k in sorted_ar if k!=None}

key_dict = {sorted_ar[i]:[my_summer(i,sorted_ar[i],key_dict),my_counter(sorted_ar[i],key_dict)] for i in range(1,len(sorted_ar))}

输出：{10: [3, 2], 12: [12, 3], 15: [6, 1], 25: [7, 1]}

Answer 4

您可以尝试这样的事情：

sorted_ar = [None,10, 10, 12, 12, 12, 15, 25]

track={}
for i,j in enumerate(sorted_ar):
    if j not in track:
        track[j]=[(i,1)]
    else:
        track[j].append((i,1))


final_={}
for i,j in track.items():
    if i not in final_:
        final_[i]=(sum(list(map(lambda x:x[0],j))),sum(list(map(lambda x:x[1],j))))

print(final_)

输出：

{None: (0, 1), 10: (3, 2), 15: (6, 1), 12: (12, 3), 25: (7, 1)}

从排序列表创建字典

4 个答案: