如何使用分组求和将值插入numpy数组

import numpy as np

arr = np.array([])

grouping_key = [1, 1, 2, 2, 2, 3, 3, 3, 3]

values = [10, 20, 30, 40, 50, 60, 70, 80, 90]

unique_keys = np.unique(grouping_key)

split_values = np.split(values, np.cumsum(np.unique(grouping_key, return_counts=True)[1])[:-1])

summed_values = np.array([np.sum(group) for group in split_values])

arr = np.concatenate([arr, summed_values])

import numpy as np

# 创建一个空的NumPy数组
arr = np.array([])

# 定义分组键和值列表
grouping_key = [1, 1, 2, 2, 2, 3, 3, 3, 3]
values = [10, 20, 30, 40, 50, 60, 70, 80, 90]

# 获取唯一的分组键
unique_keys = np.unique(grouping_key)

# 拆分值列表
split_values = np.split(values, np.cumsum(np.unique(grouping_key, return_counts=True)[1])[:-1])

# 对每个分组求和
summed_values = np.array([np.sum(group) for group in split_values])

# 插入求和结果到原始数组
arr = np.concatenate([arr, summed_values])

print(arr)