Início / user-14984570

Kazdegotepu's questions

Martin Hope
Kazdegotepu
Asked: 2024-10-11 18:13:29 +0800 CST
  • 8

Gostaria de obter valor de uma coluna cujo nome está em outra coluna.

Na minha tabela há muitas colunas, então fazer isso pl.when().then()não é uma opção.

Como entrada temos este dataframe:

foo contagem_de_foo bar contagem_de_barras base contagem_base maior
1 23 4 43 5 64 base
2 45 6 45 1 43 bar
3 234 9 453 15 231 base
4 55 2 67 3 94 foo

e gostaria de transformar isso com with_columns():

foo contagem_de_foo bar contagem_de_barras base contagem_base maior maior_contagem
1 23 4 43 5 64 base 64
2 45 6 45 1 43 bar 45
3 234 9 453 15 231 base 231
4 55 2 67 3 94 foo 4

Este (pseudo)código ilustra o que tenho em mente (mas é claro que isso não funciona)

df = pl.DataFrame({"foo" : [1, 2, 3, 4],
                   "foo_count" : [23, 45 ,234, 55],
                   "bar" : [4 ,6 ,9, 2],
                   "bar_count" : [43, 45 ,453, 67],
                   "baz": [5,1,15, 3],
                   "baz_count" : [64, 43 ,231, 94],
                   "largest" : ["baz", "bar", "baz", "foo"]})

df.with_columns(
    pl.col(f"{pl.col('largest')}_count").alias("largest_count")
)
python-polars
Martin Hope
Kazdegotepu
Asked: 2024-08-02 18:22:24 +0800 CST
  • 8

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