Início / coding / Perguntas / 79207871
Accepted
Andi
Asked: 2024-11-20 23:24:34 +0800 CST

Substituir os dois últimos valores de linha em um DataFrame de polares agrupados

  • 772

Preciso substituir os dois últimos valores na valuecoluna a pl.DataFramepor zeros, por isso preciso da group_bycoluna 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     │
└────────┴───────┘

Aqui está o meu resultado esperado:

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 respostas

  1. Best Answer

    Você pode usar

    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     │
└────────┴───────┘

    Alternativamente, você pode usar

    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. Você pode usar .is_last_distinct()e.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

relate perguntas