主页 / coding / 问题 / 79207871
Accepted
Andi
Asked: 2024-11-20 23:24:34 +0800 CST

替换分组极坐标 DataFrame 中的最后两行值

  • 772

我需要用零替换valuea 列中的最后两个值pl.DataFrame，从而我需要group_by该symbol列。

import polars as pl

df = pl.DataFrame(
    {"symbol": [*["A"] * 4, *["B"] * 4], "value": range(8)}
)

shape: (8, 2)
┌────────┬───────┐
│ symbol ┆ value │
│ ---    ┆ ---   │
│ str    ┆ i64   │
╞════════╪═══════╡
│ A      ┆ 0     │
│ A      ┆ 1     │
│ A      ┆ 2     │
│ A      ┆ 3     │
│ B      ┆ 4     │
│ B      ┆ 5     │
│ B      ┆ 6     │
│ B      ┆ 7     │
└────────┴───────┘

这是我的预期结果：

shape: (8, 2)
┌────────┬───────┐
│ symbol ┆ value │
│ ---    ┆ ---   │
│ str    ┆ i64   │
╞════════╪═══════╡
│ A      ┆ 0     │
│ A      ┆ 1     │
│ A      ┆ 0     │<-- replaced
│ A      ┆ 0     │<-- replaced
│ B      ┆ 4     │
│ B      ┆ 5     │
│ B      ┆ 0     │<-- replaced
│ B      ┆ 0     │<-- replaced
└────────┴───────┘
python

2 个回答

  1. Best Answer

    您可以使用

    df.with_columns(
    pl.col.value.head(pl.len() - 2).append(pl.repeat(0, 2))
    .over("symbol")
)

    shape: (8, 2)
┌────────┬───────┐
│ symbol ┆ value │
│ ---    ┆ ---   │
│ str    ┆ i64   │
╞════════╪═══════╡
│ A      ┆ 0     │
│ A      ┆ 1     │
│ A      ┆ 0     │
│ A      ┆ 0     │
│ B      ┆ 4     │
│ B      ┆ 5     │
│ B      ┆ 0     │
│ B      ┆ 0     │
└────────┴───────┘

    或者，你可以使用

    df.with_columns(
    pl.when(pl.int_range(pl.len()) < pl.len() - 2).then(pl.col.value)
    .otherwise(0)
    .over("symbol") 
)

    shape: (8, 2)
┌────────┬───────┐
│ symbol ┆ value │
│ ---    ┆ ---   │
│ str    ┆ i64   │
╞════════╪═══════╡
│ A      ┆ 0     │
│ A      ┆ 1     │
│ A      ┆ 0     │
│ A      ┆ 0     │
│ B      ┆ 4     │
│ B      ┆ 5     │
│ B      ┆ 0     │
│ B      ┆ 0     │
└────────┴───────┘
    • 1

  2. 您可以使用.is_last_distinct()和.shift()

    df.with_columns(
   pl.when(
      pl.any_horizontal(
          pl.col("symbol").is_last_distinct(),
          pl.col("symbol").shift(-1).over("symbol").is_last_distinct()
      ).not_()
   )
   .then(pl.col("value"))
   .otherwise(0)
)

    shape: (8, 2)
┌────────┬───────┐
│ symbol ┆ value │
│ ---    ┆ ---   │
│ str    ┆ i64   │
╞════════╪═══════╡
│ A      ┆ 0     │
│ A      ┆ 1     │
│ A      ┆ 0     │
│ A      ┆ 0     │
│ B      ┆ 4     │
│ B      ┆ 5     │
│ B      ┆ 0     │
│ B      ┆ 0     │
└────────┴───────┘
    • 0

相关问题