Python如何添加多个列表的和

2024-05-19 07:05:01 发布

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

以下是列表列表。如何将“总黑”和“总色”的总和添加到一个列表中,使“帐户ID”和“帐户名称”保持相同?你知道吗

[[{'Total Black': 3, 'Total Color': 2, 'Account ID': '2222', 'Account Name': 'Moe'}, 
  {'Total Black': 5, 'Total Color': 4, 'Account ID': '3333', 'Account Name': 'Larry'}, 
  {'Total Black': 9, 'Total Color': 6, 'Account ID': '4444', 'Account Name': 'Curly'}], 
 [{'Total Black': 8, 'Total Color': 4, 'Account ID': '2222', 'Account Name': 'Moe'}, 
  {'Total Black': 2, 'Total Color': 7, 'Account ID': '3333', 'Account Name': 'Larry'}, 
  {'Total Black': 11, 'Total Color': 9, 'Account ID': '4444', 'Account Name': 'Curly'}]]

所以结果是:

[{'Total Black': 11, 'Total Color': 6, 'Account ID': '2222', 'Account Name': 
'Moe'}, {'Total Black': 7, 'Total Color': 11, 'Account ID': '3333', 'Account 
Name': 'Larry'}, {'Total Black': 20, 'Total Color': 15, 'Account ID': 4444, 
'Account Name': 'Curly'}]

Tags: name名称id列表帐户accountcolortotal
3条回答

我们用熊猫吧。你知道吗

#From @C14L setup:

li = [
    [
        {'Total Black': 3, 'Total Color': 2, 'Account ID': '2222', 'Account Name': 'Moe'},
        {'Total Black': 5, 'Total Color': 4, 'Account ID': '3333', 'Account Name': 'Larry'},
        {'Total Black': 9, 'Total Color': 6, 'Account ID': '4444', 'Account Name': 'Curly'}
    ],
    [
        {'Total Black': 8, 'Total Color': 4, 'Account ID': '2222', 'Account Name': 'Moe'},
        {'Total Black': 2, 'Total Color': 7, 'Account ID': '3333', 'Account Name': 'Larry'},
        {'Total Black': 11, 'Total Color': 9, 'Account ID': '4444', 'Account Name': 'Curly'}
    ]
]

pd.concat([pd.DataFrame(li[i]) for i in range(len(li))])\
  .groupby(['Account ID','Account Name'])\
  .sum()\
  .reset_index()\
  .to_dict(orient='records')

输出:

[{'Account ID': '2222',
  'Account Name': 'Moe',
  'Total Black': 11,
  'Total Color': 6},
 {'Account ID': '3333',
  'Account Name': 'Larry',
  'Total Black': 7,
  'Total Color': 11},
 {'Account ID': '4444',
  'Account Name': 'Curly',
  'Total Black': 20,
  'Total Color': 15}]

详情:

首先,我们使用列表理解和pandas dataframe构造函数为带有字典的列表的原始列表的每个元素构建数据帧。你知道吗

下一步,使用pd.concat公司将这两个数据帧放在一起,在本例中是将数据帧合并为一个数据帧。你知道吗

使用groupby with sum来聚合Total Color和Total Black的值。你知道吗

最后,重设索引和todict以输出组合数据帧作为字典。你知道吗

设置

thing = [
    [{'Total Black': 3, 'Total Color': 2, 'Account ID': '2222', 'Account Name': 'Moe'}, 
     {'Total Black': 5, 'Total Color': 4, 'Account ID': '3333', 'Account Name': 'Larry'}, 
     {'Total Black': 9, 'Total Color': 6, 'Account ID': '4444', 'Account Name': 'Curly'}], 
    [{'Total Black': 8, 'Total Color': 4, 'Account ID': '2222', 'Account Name': 'Moe'}, 
     {'Total Black': 2, 'Total Color': 7, 'Account ID': '3333', 'Account Name': 'Larry'}, 
     {'Total Black': 11, 'Total Color': 9, 'Account ID': '4444', 'Account Name': 'Curly'}]
]

与ScottBoston非常相似

pd.DataFrame(sum(thing, [])).groupby(
    ['Account ID', 'Account Name'], as_index=False
).sum().to_dict('r')

merge_with

from itertools import chain
from toolz.dicttoolz import merge_with
from collections import defaultdict

idkey = ('Account ID', 'Account Name')

x = defaultdict(list)
for d in map(dict, chain(*thing)):
  x[(*map(d.pop, idkey),)].append(d)

[{**dict(zip(idkey, key)), **merge_with(sum, *d)} for key, d in x.items()]

[{'Account ID': '2222',
  'Account Name': 'Moe',
  'Total Black': 11,
  'Total Color': 6},
 {'Account ID': '3333',
  'Account Name': 'Larry',
  'Total Black': 7,
  'Total Color': 11},
 {'Account ID': '4444',
  'Account Name': 'Curly',
  'Total Black': 20,
  'Total Color': 15}]
li = [
    [
        {'Total Black': 3, 'Total Color': 2, 'Account ID': '2222', 'Account Name': 'Moe'},
        {'Total Black': 5, 'Total Color': 4, 'Account ID': '3333', 'Account Name': 'Larry'},
        {'Total Black': 9, 'Total Color': 6, 'Account ID': '4444', 'Account Name': 'Curly'}
    ],
    [
        {'Total Black': 8, 'Total Color': 4, 'Account ID': '2222', 'Account Name': 'Moe'},
        {'Total Black': 2, 'Total Color': 7, 'Account ID': '3333', 'Account Name': 'Larry'},
        {'Total Black': 11, 'Total Color': 9, 'Account ID': '4444', 'Account Name': 'Curly'}
    ]
]

result = [
    {
        'sum': x['Total Black'] + x['Total Color'],
        'Account ID': x['Account ID'],
        'Account Name': x['Account Name']
    } for x in li[0]
]

给出第一个子列表的结果

[
    {'sum': 5, 'Account ID': '2222', 'Account Name': 'Moe'},
    {'sum': 9, 'Account ID': '3333', 'Account Name': 'Larry'},
    {'sum': 15, 'Account ID': '4444', 'Account Name': 'Curly'},
]

这就是你要找的吗?你知道吗

现在您只需要遍历外部列表,对每个包含的列表重复此操作。你知道吗

相关问题 更多 >

    热门问题