以下是一个示例代码,根据一列对二维数组的行进行分组,并对另一列进行求和:
def group_and_sum(arr, column_to_group, column_to_sum):
result = {}
for row in arr:
key = row[column_to_group]
value = row[column_to_sum]
if key in result:
result[key] += value
else:
result[key] = value
return result
# 示例用法
arr = [
[1, 10],
[1, 20],
[2, 5],
[2, 15],
[3, 30],
]
# 按照第一列对第二列进行分组求和
result = group_and_sum(arr, 0, 1)
print(result)
输出结果为:{1: 30, 2: 20, 3: 30}
,表示按照第一列对第二列进行分组求和的结果。