主页 / user-14984570

Kazdegotepu's questions

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

我想从另一列中获取名称所在的列的值。

我的表中有很多列，因此这样做pl.when().then()不是一个选择。

作为输入，我们有这个数据框：

富 foo_count 酒吧 条形计数 巴兹 baz_count 最大的
1 23 4 43 5 64 巴兹
2 四十五 6 四十五 1 43 酒吧
3 234 9 453 15 231 巴兹
4 55 2 67 3 94 富

并希望将其转换为with_columns()：

富 foo_count 酒吧 条形计数 巴兹 baz_count 最大的 最大计数
1 23 4 43 5 64 巴兹 64
2 四十五 6 四十五 1 43 酒吧 四十五
3 234 9 453 15 231 巴兹 231
4 55 2 67 3 94 富 4

这个（伪）代码说明了我的想法（但当然这不起作用）

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

尽管此方法类型的文档提示参数如下，但我无法使用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