Início / coding / Perguntas / 78950520
Accepted
Phil-ZXX
Asked: 2024-09-05 04:49:48 +0800 CST

Use o especificador de formato para converter a coluna float/int no dataframe polars para string

  • 772

Eu tenho este código:

import polars as pl
df = pl.DataFrame({'size': [34.2399, 1232.22, -479.1]})
df.with_columns(pl.format('{:,.2f}', pl.col('size')))

Mas falha:

ValueError - Traceback, line 3
      2 df = pl.DataFrame({'size': [34.2399, 1232.22, -479.1]})
----> 3 df.with_columns(pl.format('{:,.2f}', pl.col('size')))

File polars\functions\as_datatype.py:718, in format(f_string, *args)
    717     msg = "number of placeholders should equal the number of arguments"
--> 718     raise ValueError(msg)

ValueError: number of placeholders should equal the number of arguments

Como posso formatar uma coluna floatou intusando um especificador de formato como '{:,.2f}'?

python

2 respostas

  1. Best Answer

    Conforme descrito por @mozway, strings de formato geral ainda não são suportadas como parte de pl.format. A solicitação de recursosprint correspondente já contém uma implementação polars agradável da formatação estilo C (mais comum) .

    Se a eficiência não for um problema muito grande (por exemplo, em análise exploratória de dados), você pode simplesmente usar pl.Expr.map_elementse recorrer à solução ingênua do Python.

    df.with_columns(
    pl.col("size").map_elements(lambda x: f"{x:,.2f}", return_dtype=pl.String)
)

    shape: (3, 1)
┌──────────┐
│ size     │
│ ---      │
│ str      │
╞══════════╡
│ 34.24    │
│ 1,232.22 │
│ -479.10  │
└──────────┘
    • 2

  2. pl.formatestá reconhecendo literais {}, diferentemente das f-strings do Python (se você executar df.with_columns(pl.format('{:,.2f}'))verá que {:,.2f}não foi alterado).

    Portanto, você não pode usar pl.formatda maneira que deseja (conforme postado como comentário na pergunta, esta é uma solicitação de recurso ).

    Talvez você queira usar um dos métodos descritos aqui .

    Código pl.formatque mostra como funciona, ele simplesmente divide as strings em {}( f_string.split("{}")) e as une novamente com as expressões intermediárias:

    def format(f_string: str, *args: Expr | str) -> Expr:
    if f_string.count("{}") != len(args):
        msg = "number of placeholders should equal the number of arguments"
        raise ValueError(msg)

    exprs = []

    arguments = iter(args)
    for i, s in enumerate(f_string.split("{}")):
        if i > 0:
            e = wrap_expr(parse_into_expression(next(arguments)))
            exprs.append(e)

        if len(s) > 0:
            exprs.append(F.lit(s))

    return concat_str(exprs, separator="")
    • -1

relate perguntas