當前位置: 首頁>>編程示例 >>用法及示例精選 >>正文


Python pyspark Rolling.mean用法及代碼示例

本文簡要介紹 pyspark.pandas.window.Rolling.mean 的用法。

用法:

Rolling.mean() → FrameLike

計算值的滾動平均值。

注意

此 API 的當前實現使用 Spark 的 Window 而不指定分區規範。這會導致將所有數據移動到單個機器中的單個分區中,並可能導致嚴重的性能下降。避免對非常大的數據集使用此方法。

返回

係列或DataFrame

返回的對象類型由滾動計算的調用者確定。

例子

>>> s = ps.Series([4, 3, 5, 2, 6])
>>> s
0    4
1    3
2    5
3    2
4    6
dtype: int64
>>> s.rolling(2).mean()
0    NaN
1    3.5
2    4.0
3    3.5
4    4.0
dtype: float64
>>> s.rolling(3).mean()
0         NaN
1         NaN
2    4.000000
3    3.333333
4    4.333333
dtype: float64

對於 DataFrame,每個滾動平均值都是按列計算的。

>>> df = ps.DataFrame({"A": s.to_numpy(), "B": s.to_numpy() ** 2})
>>> df
   A   B
0  4  16
1  3   9
2  5  25
3  2   4
4  6  36
>>> df.rolling(2).mean()
     A     B
0  NaN   NaN
1  3.5  12.5
2  4.0  17.0
3  3.5  14.5
4  4.0  20.0
>>> df.rolling(3).mean()
          A          B
0       NaN        NaN
1       NaN        NaN
2  4.000000  16.666667
3  3.333333  12.666667
4  4.333333  21.666667

相關用法


注:本文由純淨天空篩選整理自spark.apache.org大神的英文原創作品 pyspark.pandas.window.Rolling.mean。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。