主页 / coding / 问题 / 79566761
Accepted
Ghost
Asked: 2025-04-10 21:19:01 +0800 CST

根据外部列表对极坐标数据框进行排序

  • 772

早晨，

我不确定这是否能实现。

假设我有一个带有列 a、b（无论什么）的极点数据框。

df = pl.DataFrame({"a":[1,2,3,4,5],"b":['x','y','z','p','f']})

还有一个列表.. l = [1,3,5,2,4]；是否可以使用列表 l 作为排序顺序对数据框（使用列“a”）进行排序？

提前致谢！

python

2 个回答

  1. Best Answer

    您可以使用Enum自sort定义顺序，但由于Enum仅适用于字符串，因此您首先需要临时转换为字符串：

    df.sort(by=pl.col('a').cast(pl.String).cast(pl.Enum(list(map(str, l)))))

    输出：

    ┌─────┬─────┐
│ a   ┆ b   │
│ --- ┆ --- │
│ i64 ┆ str │
╞═════╪═════╡
│ 1   ┆ x   │
│ 3   ┆ z   │
│ 5   ┆ f   │
│ 2   ┆ y   │
│ 4   ┆ p   │
└─────┴─────┘
    • 2

  2. 你不能只关注.replace()价值观吗？

    df.sort(pl.col("a").replace(l, range(len(l))))

    shape: (5, 2)
┌─────┬─────┐
│ a   ┆ b   │
│ --- ┆ --- │
│ i64 ┆ str │
╞═════╪═════╡
│ 1   ┆ x   │
│ 3   ┆ z   │
│ 5   ┆ f   │
│ 2   ┆ y   │
│ 4   ┆ p   │
└─────┴─────┘

    如果只是等长的 1:1 映射，你也可以.sort_by()

    df.sort(pl.col("a").sort_by(pl.Series(l)))
    • 2

相关问题