RDD 的 Pyspark 平均间隔

Question

我正在尝试使用 PySpark 查找相邻元组列表之间的平均差异。

例如，如果我有这样的 RDD

vals = [(2,110),(2,130),(2,120),(3,200),(3,206),(3,206),(4,150),(4,160),(4,170)]

我想找出每个键的平均差异。

例如键值“2”

平均差为 (abs(110-130) + abs(130-120))/2 = 15。

这是我目前的方法。我正在尝试更改平均计算代码以适应这种情况。但是好像不行。

from pyspark import SparkContext
aTuple = (0,0)
interval = vals.aggregateByKey(aTuple, lambda a,b: (abs(a[0] - b),a[1] + 1),
                                       lambda a,b: (a[0] + b[0], a[1] + b[1]))
finalResult = interval.mapValues(lambda v: (v[0]/v[1])).collect()

我想使用 RDD 函数来做到这一点，没有 Spark SQL 或任何其他附加包。

最好的方法是什么？

如果您有任何问题，请告诉我。

感谢您的宝贵时间。

Answer 1

我想出了一个天真的方法。我不确定这是否适用于所有情况。它是这样的。

我们先做一个计算移动平均线的函数。如果这不是计算移动平均线的正确方法，请指正。

def get_abs(num_list):
    '''
    >>> get_abs([110, 130, 120])
    15.0
    '''
    acc = 0
    num_pairs = 0
    for i in range(len(num_list)-1):
        acc += abs(num_list[i]-num_list[i+1])
        num_pairs +=1
    return acc/num_pairs

接下来，我们将列表并行化

>>> vals = [(2,110),(2,130),(2,120),(3,200),(3,206),(3,206),(4,150),(4,160),(4,170)]
>>> rdd = sc.parallelize(vals)
>>> rdd.collect()
[(2, 110),
 (2, 130),
 (2, 120),
 (3, 200),
 (3, 206),
 (3, 206),
 (4, 150),
 (4, 160),
 (4, 170)]

然后，将属于同一列表的值分组。

>>> vals = rdd.groupByKey().mapValues(list)
>>> vals.collect()
[(4, [150, 160, 170]), (2, [110, 130, 120]), (3, [200, 206, 206])]

然后我们只需要调用我们上面定义的函数来计算分组值的移动平均值。

>>> vals.mapValues(get_abs).collect()
[(4, 10.0), (2, 15.0), (3, 3.0)]

RDD 的 Pyspark 平均间隔

Pyspark Average interval for RDD

python

moving-average

apache-spark

rdd

pyspark