如何在Polars Rust中为DataFrame添加列/Series?
在Polars中添加列的正确姿势:别再死磕append/extend了
我之前在给Polars DataFrame加列这个简单事儿上耗了好长时间,试了append、extend各种方法都没搞定,最后才搞明白,Polars里要加列得用with_column方法。
给大家举个实际例子:
先搞个基础的DataFrame:
import polars as pl df = pl.DataFrame({ "name": ["Alice", "Bob", "Charlie"], "age": [25, 30, 35] })
要是想加一列is_adult,判断年龄是否满18,用with_column这么写就行:
df = df.with_column((pl.col("age") >= 18).alias("is_adult"))
要是想一次加多列,还能用复数形式的with_columns:
df = df.with_columns( (pl.col("age") >= 18).alias("is_adult"), (pl.col("age") + 5).alias("age_in_5_years") )
之前踩坑完全是把Python列表的方法套到Polars上了——Polars的append是用来合并整个DataFrame的,根本不是加单列的;extend也不是干这个的,方向直接错了。
内容的提问来源于stack exchange,提问作者nklsla
相关产品推荐
相关产品推荐

