Início / coding / Perguntas / 79566761
Accepted
Ghost
Asked: 2025-04-10 21:19:01 +0800 CST

Classificar um dataframe polar com base em uma lista externa

  • 772

Manhã,

Não tenho certeza se isso pode ser alcançado.

Digamos que eu tenha um dataframe polar com colunas a, b (qualquer que seja).

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

E uma lista... l = [1,3,5,2,4]; é possível classificar o dataframe (usando a coluna "a") usando a lista l como ordem de classificação?

Desde já, obrigado!

python

2 respostas

  1. Best Answer

    Você pode usar um Enumto sortcom uma ordem personalizada, porém, como Enumsó funciona com strings, primeiro você precisa converter temporariamente para string:

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

    Saída:

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

  2. Você não pode apenas .replace()os valores?

    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   │
└─────┴─────┘

    Se for apenas um mapeamento 1:1 de comprimentos iguais, você também pode.sort_by()

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

relate perguntas