Início / coding / Perguntas / 79181689
Accepted
Samuel Allain
Asked: 2024-11-12 23:11:23 +0800 CST

polars `read_csv()` para ler de uma string e não de um arquivo

  • 772

é possível ler de uma string com pl.read_csv()? Algo assim, que funcionaria:

content = """c1, c2
            A,1
            B,3
            C,2"""
pl.read_csv(content)

Eu sei disso, é claro:

pl.DataFrame({"c1":["A", "B", "C"],"c2" :[1,3,2]})

Mas é propenso a erros com tabelas longas e você tem que contar números para saber qual valor modificar. Eu também sei sobre dicionários, mas tenho mais de 2 colunas no meu exemplo da vida real.

Contexto: Eu costumava usar fread()o R ​​data.table e era muito útil, especialmente quando você quer converter uma coluna com a ajuda de uma junção, em vez de instruções ifelse() complicadas.

Obrigado !

python

1 respostas

  1. Best Answer

    pl.read_csv()aceita IO como parâmetro de origem .

    fonte: str | Caminho | IO[str] | IO[bytes] | bytes

    Então você pode usar io.StringIO:

    from io import StringIO

content = """
c1,c2
A,1
B,3
C,2
"""

data = StringIO(content)
pl.read_csv(data)

    shape: (3, 2)
┌─────┬─────┐
│ c1  ┆ c2  │
│ --- ┆ --- │
│ str ┆ i64 │
╞═════╪═════╡
│ A   ┆ 1   │
│ B   ┆ 3   │
│ C   ┆ 2   │
└─────┴─────┘

    Como você pode ver acima, você também pode passar bytes como parâmetro de origem . Você pode usar str.encode()o método para isso:

    content = """
c1,c2
A,1
B,3
C,2
"""

pl.read_csv(content.encode())

    shape: (3, 2)
┌─────┬─────┐
│ c1  ┆ c2  │
│ --- ┆ --- │
│ str ┆ i64 │
╞═════╪═════╡
│ A   ┆ 1   │
│ B   ┆ 3   │
│ C   ┆ 2   │
└─────┴─────┘
    • 5

relate perguntas