Início / coding / Perguntas / 78847998
Accepted
Rhys
Asked: 2024-08-08 18:54:18 +0800 CST

Listar para DataFrame com cabeçalhos de linha e coluna

  • 772

Preciso converter uma lista (incluindo cabeçalhos) em um Dataframe.

Se eu fizer isso diretamente usando pl.DataFrame(list), os cabeçalhos são criados e tudo é mantido como uma string. Além disso, a tabela é transposta, de forma que o primeiro elemento da lista se torne a primeira coluna do dataframe.

Lista de entrada.

[
    ['Earnings estimate', 'Current qtr. (Jun 2024)', 'Next qtr. (Sep 2024)', 'Current year (2024)', 'Next year (2025)'],
    ['No. of analysts', '13', '11', '26', '26'],
    ['Avg. Estimate', '1.52', '1.62', '6.27', '7.23'],
    ['Low estimate', '1.36', '1.3', '5.02', '5.88'],
    ['High estimate', '1.61', '1.74', '6.66', '8.56'],
    ['Year ago EPS', '1.76', '1.36', '5.74', '6.27'],
]

Resultado esperado.

Resultado Esperado

python

3 respostas

  1. Best Answer

    Você pode definir explicitamente orient=para evitar a transposição:

    pl.DataFrame(data[1:], orient="row", schema=data[0])

    shape: (5, 5)
┌───────────────────┬─────────────────────────┬──────────────────────┬─────────────────────┬──────────────────┐
│ Earnings estimate ┆ Current qtr. (Jun 2024) ┆ Next qtr. (Sep 2024) ┆ Current year (2024) ┆ Next year (2025) │
│ ---               ┆ ---                     ┆ ---                  ┆ ---                 ┆ ---              │
│ str               ┆ str                     ┆ str                  ┆ str                 ┆ str              │
╞═══════════════════╪═════════════════════════╪══════════════════════╪═════════════════════╪══════════════════╡
│ No. of analysts   ┆ 13                      ┆ 11                   ┆ 26                  ┆ 26               │
│ Avg. Estimate     ┆ 1.52                    ┆ 1.62                 ┆ 6.27                ┆ 7.23             │
│ Low estimate      ┆ 1.36                    ┆ 1.3                  ┆ 5.02                ┆ 5.88             │
│ High estimate     ┆ 1.61                    ┆ 1.74                 ┆ 6.66                ┆ 8.56             │
│ Year ago EPS      ┆ 1.76                    ┆ 1.36                 ┆ 5.74                ┆ 6.27             │
└───────────────────┴─────────────────────────┴──────────────────────┴─────────────────────┴──────────────────┘
    • 2

  2. Para completar, Polars também possui um construtor para dados semelhantes a registros: polars.from_records que você também pode usar aqui.

    import polars as pl

data = [
    ['Earnings estimate',  'Current qtr.',  'Next qtr.'],
    ['No. of analysts',    '13',            '11',      ],
    ['Avg. Estimate',      '1.52',          '1.62',    ],
    ['Low estimate',       '1.36',          '1.3',     ],
    ['High estimate',      '1.61',          '1.74',    ],
    ['Year ago EPS',       '1.76',          '1.36',    ],
]

df = pl.from_records(data[1:], schema=data[0])
print(df)
# shape: (5, 3)
# ┌───────────────────┬──────────────┬───────────┐
# │ Earnings estimate ┆ Current qtr. ┆ Next qtr. │
# │ ---               ┆ ---          ┆ ---       │
# │ str               ┆ str          ┆ str       │
# ╞═══════════════════╪══════════════╪═══════════╡
# │ No. of analysts   ┆ 13           ┆ 11        │
# │ Avg. Estimate     ┆ 1.52         ┆ 1.62      │
# │ Low estimate      ┆ 1.36         ┆ 1.3       │
# │ High estimate     ┆ 1.61         ┆ 1.74      │
# │ Year ago EPS      ┆ 1.76         ┆ 1.36      │
# └───────────────────┴──────────────┴───────────┘
    • 2

  3. DataFrameconstrutor tem um parâmetro orient:

    orient : {'col', 'row'}, default Nenhum Se deve interpretar dados bidimensionais como colunas ou linhas. Se Nenhum, a orientação será inferida pela correspondência entre colunas e dimensões de dados. Se isso não produzir resultados conclusivos, a orientação da coluna será usada.

    Portanto, no seu caso, cola orientação é usada, mas você pode configurá-la manualmente para row:

    pl.DataFrame(data[1:], schema=data[0], orient="row")

    shape: (5, 5)
┌───────────────────┬─────────────────────────┬──────────────────────┬─────────────────────┬──────────────────┐
│ Earnings estimate ┆ Current qtr. (Jun 2024) ┆ Next qtr. (Sep 2024) ┆ Current year (2024) ┆ Next year (2025) │
│ ---               ┆ ---                     ┆ ---                  ┆ ---                 ┆ ---              │
│ str               ┆ str                     ┆ str                  ┆ str                 ┆ str              │
╞═══════════════════╪═════════════════════════╪══════════════════════╪═════════════════════╪══════════════════╡
│ No. of analysts   ┆ 13                      ┆ 11                   ┆ 26                  ┆ 26               │
│ Avg. Estimate     ┆ 1.52                    ┆ 1.62                 ┆ 6.27                ┆ 7.23             │
│ Low estimate      ┆ 1.36                    ┆ 1.3                  ┆ 5.02                ┆ 5.88             │
│ High estimate     ┆ 1.61                    ┆ 1.74                 ┆ 6.66                ┆ 8.56             │
│ Year ago EPS      ┆ 1.76                    ┆ 1.36                 ┆ 5.74                ┆ 6.27             │
└───────────────────┴─────────────────────────┴──────────────────────┴─────────────────────┴──────────────────┘
    • 1

relate perguntas