将Dict项添加到列表中,并将具有相同键的值附加到较新的列表项中

2024-06-14 11:20:50 发布

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

我有以下实现:

from collections import defaultdict
from collections import OrderedDict

prod = [
    [1, 'tomato', 'veg', 'Jan-1'],
    [1, 'banana', 'fruit', 'Jan-3'],
    [2, 'melon', 'fruit', 'Jan-2'],
    [3, 'apple', 'fruit', 'Jan-4'],
    [2, 'cucumber', 'veg', 'Jan-1']
]

d = defaultdict(list)

for i in range (0, len(prod)):
    f_name = prod[i][1]
    f_type = prod[i][2]
    f_date = prod[i][3]

    key = prod[i][0]

    d[key].append([f_name, f_type, f_date])

e = OrderedDict(sorted(d.items(), key=lambda t: t[0]))
print ("***************")
print (e)

table_for_graph = []

for key, value in e.iteritems():
    table_for_graph.append(value)

print (table_for_graph)

我得到的输出是这样的:

[[['tomato', 'veg', 'Jan-1'], ['banana', 'fruit', 'Jan-3']], [['melon', 'fruit', 'Jan-2'], ['cucumber', 'veg', 'Jan-1']], [['apple', 'fruit', 'Jan-4']]]

我想创建这样一个列表:

[
    ['tomato''\n''banana','veg''\n''fruit','Jan-1''\n''Jan-3'],
    ['melon''\n''cucumber','fruit''\n''veg','Jan-2''\n''Jan-1'],
    ['apple','fruit','Jan-4']
]

意思是,我想连接具有相同键的项。 我该怎么办?我还不熟悉迭代thorugh dict


Tags: keyfromapplefortableprodjangraph
1条回答
网友
1楼 · 发布于 2024-06-14 11:20:50

对于初学者来说,构造字典的循环可以大大简化,因为解包和重新打包是多余的,除非您需要对元素重新排序:

for item in prod:
    d[item[0]] = item[1:]

其次,table_graph最初根本不需要循环:

table_for_graph = e.values()

这里的每个项目都是一个嵌套列表,如:

>>> e[1]
[['tomato', 'veg', 'Jan-1'],
 ['banana', 'fruit', 'Jan-3']]

通过将位压缩在一起,可以有效地进行转置:

>>> zip(*e[1])
[['tomato', 'banana'],
 ['veg', 'fruit'],
 ['Jan-1', 'Jan-3']]

综合起来:

 >>> [s in line for line in zip(*e[1]) for s in line]
 ['tomato', 'banana', 'veg', 'fruit', 'Jan-1', 'Jan-3']

可以在字典的每个元素上运行该表达式:

table_for_graph = [[s for line in zip(*value) for s in line] for value in e.itervalues()]

相关问题 更多 >