Início / coding / Perguntas / 78824984
Accepted
Kazdegotepu
Asked: 2024-08-02 18:22:24 +0800 CST

Não é possível usar o seletor em polars.DataFrame.unpivot()

  • 772

Não consigo usar o pl.exclude()seletor como índice, pl.DataFrame.unpivotapesar da documentação desse tipo de método sugerir o indexparâmetro da seguinte maneira.

index: 'ColumnNameOrSelector | Sequence[ColumnNameOrSelector] | None' = None

O código a seguir pode ser usado para reproduzir o erro.

import polars as pl

df = pl.DataFrame({
    'foo' : ['one', 'two', 'three'],
    'bar' : ['four', 'five', 'six'],
    'baz' : [10, 20, 30],
    'qux' : [50, 60, 70]
})

column_list = ['baz', 'qux']
df.unpivot(column_list, index=pl.exclude(column_list))

TypeError: argument 'index': 'Expr' object cannot be converted to 'PyString'

Além disso, usar as colunas explicitamente funciona bem.

df.unpivot(column_list, index=['foo', 'bar'])

Então, como os seletores de coluna podem ser usados pl.DataFrame.unpivot?

python

1 respostas

  1. Best Answer

    pl.excludeé uma expressão polar "regular" (do tipo pl.Expr). Seletores são objetos especiais (também expressões, mas de um subtipo específico cs._selector_proxy_) que podem ser encontrados em arquivos pl.selectors.

    import polars.selectors as cs

column_list = ['baz', 'qux']
df.unpivot(column_list, index=cs.exclude(column_list))

    shape: (6, 4)
┌───────┬──────┬──────────┬───────┐
│ foo   ┆ bar  ┆ variable ┆ value │
│ ---   ┆ ---  ┆ ---      ┆ ---   │
│ str   ┆ str  ┆ str      ┆ i64   │
╞═══════╪══════╪══════════╪═══════╡
│ one   ┆ four ┆ baz      ┆ 10    │
│ two   ┆ five ┆ baz      ┆ 20    │
│ three ┆ six  ┆ baz      ┆ 30    │
│ one   ┆ four ┆ qux      ┆ 50    │
│ two   ┆ five ┆ qux      ┆ 60    │
│ three ┆ six  ┆ qux      ┆ 70    │
└───────┴──────┴──────────┴───────┘
    • 4

relate perguntas