從字典中添加 RDD Python 中的 NumpyArray 值

Question

如何分別在 RDD 中添加字典中的值？

我有以下內容：

my_model = sc.parallelize([("Smith", np.array([1, 0, 3])), ("Johnson",np.array([2, 4, 1])), ("Brown",np.array([1, 1, 2]))])

我希望最終的 output 成為 numpy 數組中的以下內容：

[4, 5, 6]

如何在 numpy 數組中添加這個 output？

Answer 1

您可以使用reduce function 添加 RDD 中的值：

from operator import add
result = my_model.map(lambda x: x[1]).reduce(add)