如何在Python的RDD中对Numpy数组值分别求和得到结果数组？

要得到目标求和结果，只需对RDD做两步处理：

1. 提取所有Numpy数组
   通过map操作过滤掉键（姓名），只保留每个元素中的Numpy数组：

   arrays_rdd = my_model.map(lambda x: x[1])
   

2. 归约求和
   利用reduce操作将所有数组逐元素相加，Numpy原生支持数组间的逐元素加法，直接用lambda表达式实现合并逻辑：

   sum_array = arrays_rdd.reduce(lambda a, b: a + b)
   

也可以将两步合并为一行代码：

sum_array = my_model.map(lambda x: x[1]).reduce(lambda a, b: a + b)

执行后打印sum_array，就能得到目标输出：

print(sum_array)
# 输出：[4 5 6]

内容的提问来源于stack exchange，提问作者Joe Lee