按照一个列对二维数组进行分组,并在每个组中累加另一个列的值。
创始人
2024-08-26 00:01:39
0

以下是一个示例代码,演示如何按照一个列对二维数组进行分组,并在每个组中累加另一个列的值。

from collections import defaultdict

def group_and_sum(arr, group_col, sum_col):
    groups = defaultdict(int)
    
    for row in arr:
        group = row[group_col]
        value = row[sum_col]
        groups[group] += value
        
    return groups

# 示例数据
data = [
    [1, 10],
    [2, 20],
    [1, 30],
    [2, 40],
    [3, 50]
]

grouped_data = group_and_sum(data, 0, 1)
print(grouped_data)

输出结果:

defaultdict(, {1: 40, 2: 60, 3: 50})

在这个示例中,我们定义了一个名为group_and_sum的函数,它接受一个二维数组arr、一个用于分组的列索引group_col和一个用于累加的列索引sum_col作为参数。

然后,我们使用defaultdict创建一个名为groups的字典,用于存储每个组的累加值。defaultdict是一个特殊的字典,当访问一个不存在的键时,它会自动创建一个默认值。

接下来,我们遍历二维数组中的每一行。对于每一行,我们提取分组列和累加列的值,并将累加值添加到相应组的累加值中。

最后,我们返回包含每个组的累加值的字典。

在示例数据中,我们使用第一列作为分组列,第二列作为累加列。所以我们按照第一列对数组进行分组,然后在每个组中累加第二列的值。输出结果显示了每个组的累加值。

相关内容

热门资讯

安卓换鸿蒙系统会卡吗,体验流畅... 最近手机圈可是热闹非凡呢!不少安卓用户都在议论纷纷,说鸿蒙系统要来啦!那么,安卓手机换上鸿蒙系统后,...
app安卓系统登录不了,解锁登... 最近是不是你也遇到了这样的烦恼:手机里那个心爱的APP,突然就登录不上了?别急,让我来帮你一步步排查...
安卓系统拦截短信在哪,安卓系统... 你是不是也遇到了这种情况:手机里突然冒出了很多垃圾短信,烦不胜烦?别急,今天就来教你怎么在安卓系统里...
安卓系统要维护多久,安卓系统维... 你有没有想过,你的安卓手机里那个陪伴你度过了无数日夜的安卓系统,它究竟要陪伴你多久呢?这个问题,估计...
windows官网系统多少钱 Windows官网系统价格一览:了解正版Windows的购买成本Windows 11官方价格解析微软...
安卓系统如何卸载app,轻松掌... 手机里的App越来越多,是不是感觉内存不够用了?别急,今天就来教你怎么轻松卸载安卓系统里的App,让...
怎么复制照片安卓系统,操作步骤... 亲爱的手机控们,是不是有时候想把自己的手机照片分享给朋友,或者备份到电脑上呢?别急,今天就来教你怎么...
安卓系统应用怎么重装,安卓应用... 手机里的安卓应用突然罢工了,是不是让你头疼不已?别急,今天就来手把手教你如何重装安卓系统应用,让你的...
iwatch怎么连接安卓系统,... 你有没有想过,那款时尚又实用的iWatch,竟然只能和iPhone好上好?别急,今天就来给你揭秘,怎...
iphone系统与安卓系统更新... 最近是不是你也遇到了这样的烦恼?手机更新系统总是失败,急得你团团转。别急,今天就来给你揭秘为什么iP...