rust 如何在polars Dataframe 中重复特定次数的每一行？

vlju58qv 于 2023-06-06 发布在其他

关注(0)|答案(1)|浏览(224)

let mut df = df! (
    "start" => &[6, 3, 1],
    "end" => &[7, 4, 2],
    "repeats" => &[1, 0, 2],
)?.sort(["start", "end"], false)?;

如何按重复列中给出的数字重复每一行？
下面是一个示例尝试：

df.with_column(
    all().repeat_by(col("repeats")).explode()
);

它导致了错误

the trait `AsRef<str>` is not implemented for `Expr`

1条答案

df.clone().lazy().select(
    [
        all().repeat_by(col("repeats")).explode().drop_nulls(),
    ]
);

如果你想重复另一个集合，使用lit：

all().repeat_by(lit(_series))...