Python Polars 有效枚举新列中的行

Question

我想在我的 DataFrame 中创建一个新列，枚举其中的行。我的 DataFrame 有几百万个元素，因此我需要在 Rust 引擎中处理它，所以不使用类似

apply()

的东西

我的数据框：

┌─────┬─────┬─────┐
│ foo ┆ N   ┆ bar │
│ --- ┆ --- ┆ --- │
│ str ┆ i64 ┆ str │
╞═════╪═════╪═════╡
│ A   ┆ 1   ┆ k   │
│ A   ┆ 2   ┆ l   │
│ B   ┆ 2   ┆ m   │
│ B   ┆ 4   ┆ n   │
│ C   ┆ 2   ┆ o   │
└─────┴─────┴─────┘

预期输出：

┌─────┬─────┬─────┬─────┐
│ foo ┆ N   ┆ bar ┆ c   │
│ --- ┆ --- ┆ --- ┆ --- │
│ str ┆ i64 ┆ str ┆ i64 │
╞═════╪═════╪═════╪═════╡
│ A   ┆ 1   ┆ k   ┆ 1   │
│ A   ┆ 2   ┆ l   ┆ 2   │
│ B   ┆ 2   ┆ m   ┆ 3   │
│ B   ┆ 4   ┆ n   ┆ 4   │
│ C   ┆ 2   ┆ o   ┆ 5   │
└─────┴─────┴─────┴─────┘