主页 / coding / 问题 / 78824984
Accepted
Kazdegotepu
Asked: 2024-08-02 18:22:24 +0800 CST

无法在 polars.DataFrame.unpivot() 中使用选择器

  • 772

尽管此方法类型的文档提示参数如下，但我无法使用pl.exclude()选择器作为索引。pl.DataFrame.unpivotindex

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

以下代码可用于重现该错误。

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'

此外，明确使用列也可以正常工作。

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

那么，如何使用列选择器pl.DataFrame.unpivot？

python

1 个回答

  1. Best Answer

    pl.exclude是“常规”极坐标表达式（类型）。选择器是可以在中找到的pl.Expr特殊对象（也是表达式，但属于特定子类型） 。cs._selector_proxy_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

相关问题