GH KIM

Asked: 2025-04-07 09:38:11 +0800 CST2025-04-07 09:38:11 +0800 CST 2025-04-07 09:38:11 +0800 CST

Usando uma lista de valores para selecionar linhas do Polars DataFrame

Tenho um Polars DataFrame abaixo:

import polars as pl
df = pl.DataFrame({"a":[1, 2, 3], "b":[4, 3, 2]})

>>> df
  a    b
i64  i64
  1    4
  2    3
  3    2

Posso subdividir com base em um valor específico:

x = df[df["a"] == 3]

>>> x
  a    b
i64  i64
  3    2

Mas como posso subdividir com base em uma lista de valores? - algo assim:

list_of_values = [1, 3]
y = df[df["a"] in list_of_values]

Obter:

Best Answer

Subir Chowdhury

y = df.filter(pl.col("a").is_in(list_of_values))

Saída:

┌─────┬─────┐
│ a   ┆ b   │
│ --- ┆ --- │
│ i64 ┆ i64 │
╞═════╪═════╡
│ 1   ┆ 4   │
│ 3   ┆ 2   │
└─────┴─────┘

relate perguntas