主页 / coding / 问题 / 79028838
Accepted
AColoredReptile
Asked: 2024-09-27 03:22:12 +0800 CST

极坐标滚动平均值，用空值填充窗口的开头，而不是缩短窗口

  • 772

我的问题是，在极坐标系中，是否有办法在滚动窗口开始时让值为空，直到可以填满整个窗口。例如：

dates = [
    "2020-01-01",
    "2020-01-02",
    "2020-01-03",
    "2020-01-04",
    "2020-01-05",
    "2020-01-06",
    "2020-01-01",
    "2020-01-02",
    "2020-01-03",
    "2020-01-04",
    "2020-01-05",
    "2020-01-06",
]
df = pl.DataFrame({"dt": dates, "a": [3, 4, 2, 8, 10, 1, 1, 7, 5, 9, 2, 1], "b": ["Yes","Yes","Yes","Yes","Yes", "Yes", "No", "No", "No", "No", "No", "No"]}).with_columns(
    pl.col("dt").str.strptime(pl.Date).set_sorted()
)
df = df.sort(by = 'dt')

df.rolling(
    index_column="dt", period="2d", group_by = 'b'
).agg(pl.col("a").mean().alias("ma_2d"))

结果

b   dt  ma_2d
str date    f64
"Yes"   2020-01-01  3.0
"Yes"   2020-01-02  3.5
"Yes"   2020-01-03  3.0
"Yes"   2020-01-04  5.0
"Yes"   2020-01-05  9.0

在这种情况下，我的预期是第一天应该为空，因为没有两天来填充窗口。但 polars 似乎只是截断窗口以填充起始日期。

python

2 个回答

  1. 你能检查一下长度吗？

    (df.rolling(index_column="dt", period="2d", group_by="b")
   .agg(
      pl.when(pl.len() > 1)
        .then(pl.col("a").mean())
        .alias("ma_2d")
   )
)

    shape: (12, 3)
┌─────┬────────────┬───────┐
│ b   ┆ dt         ┆ ma_2d │
│ --- ┆ ---        ┆ ---   │
│ str ┆ date       ┆ f64   │
╞═════╪════════════╪═══════╡
│ Yes ┆ 2020-01-01 ┆ null  │
│ Yes ┆ 2020-01-02 ┆ 3.5   │
│ Yes ┆ 2020-01-03 ┆ 3.0   │
│ Yes ┆ 2020-01-04 ┆ 5.0   │
│ Yes ┆ 2020-01-05 ┆ 9.0   │
│ …   ┆ …          ┆ …     │
│ No  ┆ 2020-01-02 ┆ 4.0   │
│ No  ┆ 2020-01-03 ┆ 6.0   │
│ No  ┆ 2020-01-04 ┆ 7.0   │
│ No  ┆ 2020-01-05 ┆ 5.5   │
│ No  ┆ 2020-01-06 ┆ 1.5   │
└─────┴────────────┴───────┘

    或者，有一种专用的.rolling_mean_by()方法来支持min_periods。

    df.with_columns(
   pl.col("a").rolling_mean_by("dt", window_size="2d", min_periods=2)
     .over("b")
     .alias("ma_2d")
)

    shape: (12, 4)
┌────────────┬─────┬─────┬───────┐
│ dt         ┆ a   ┆ b   ┆ ma_2d │
│ ---        ┆ --- ┆ --- ┆ ---   │
│ date       ┆ i64 ┆ str ┆ f64   │
╞════════════╪═════╪═════╪═══════╡
│ 2020-01-01 ┆ 3   ┆ Yes ┆ null  │
│ 2020-01-01 ┆ 1   ┆ No  ┆ null  │
│ 2020-01-02 ┆ 4   ┆ Yes ┆ 3.5   │
│ 2020-01-02 ┆ 7   ┆ No  ┆ 4.0   │
│ 2020-01-03 ┆ 2   ┆ Yes ┆ 3.0   │
│ …          ┆ …   ┆ …   ┆ …     │
│ 2020-01-04 ┆ 9   ┆ No  ┆ 7.0   │
│ 2020-01-05 ┆ 10  ┆ Yes ┆ 9.0   │
│ 2020-01-05 ┆ 2   ┆ No  ┆ 5.5   │
│ 2020-01-06 ┆ 1   ┆ Yes ┆ 5.5   │
│ 2020-01-06 ┆ 1   ┆ No  ┆ 1.5   │
└────────────┴─────┴─────┴───────┘
    • 1
  2. Best Answer

    有一个功能请求（#12798）来实现具有此效果的min_periods/参数，这也在问题#12049中进行了讨论。min_samples

    • 0

相关问题