Início / coding / Perguntas / 79561195
Accepted
GH KIM
Asked: 2025-04-08 12:46:27 +0800 CST

Eliminar linhas com todos os zeros no Polars DataFrame

  • 772

Posso usar drop_nans()a função para remover linhas com algumas ou todas as colunas definidas como nan.

Existe uma função equivalente para descartar linhas com todas as colunas tendo valor 0?

import polars as pl
df = pl.DataFrame({"a":[0, 0, 0, 0, 30], 
                   "b":[0, 0, 0, 0, 40],
                   "c":[0, 0, 0, 0, 50]})

>>> df

     a        b        c
   i64      i64      i64
  ------------------------
     0        0        0
     0        0        0
     0        0        0
     0        0        0
    30       40       50

Neste exemplo, gostaria de remover as primeiras 4 linhas do dataframe.

python

1 respostas

  1. Best Answer

    Você pode removecom pl.all_horizontal, aplicado à condição em pl.all:

    df.remove(pl.all_horizontal(pl.all() == 0))

    Saída:

    shape: (1, 3)
┌─────┬─────┬─────┐
│ a   ┆ b   ┆ c   │
│ --- ┆ --- ┆ --- │
│ i64 ┆ i64 ┆ i64 │
╞═════╪═════╪═════╡
│ 30  ┆ 40  ┆ 50  │
└─────┴─────┴─────┘

    Também é possível via filterinverso:

    df.filter(~pl.all_horizontal(pl.all() == 0))

# same output

    Para remover linhas com pelo menos um 0, use pl.any_horizontal:

    df = pl.DataFrame({"a":[1, 0, 0, 0, 30],
                   "b":[0, 1, 0, 0, 40],
                   "c":[0, 0, 1, 0, 50]})

df.remove(pl.any_horizontal(pl.all() == 0))

# same output
    • 3

relate perguntas