Início / coding / Perguntas / 77441311
Accepted
datahack
Asked: 2023-11-08 04:36:28 +0800 CST

O esquema Polars quebra com o tipo List

  • 772

Tentei criar um dataframe polar simples com duas colunas:

import polars as pl

data = {"a": [ ["X"], ["Y"], []], "b": [3, 4, 5]}

# this works normaly
df = pl.DataFrame(data)

df_schema = [("a", pl.List),
             ("b", pl.Int8)]

# this breaks - invalid series dtype: expected `Utf8`, got `null`
df = pl.DataFrame(data, schema=schema)

Sem especificar o esquema, ele cria o seguinte formato do Dataframe:

shape: (3, 2)
a           b
list[str]   i64
["X"]       3
["Y"]       4
[]          5

mas quando eu especifico exatamente o mesmo esquema, ele quebra. Qual poderia ser o problema?

usando polares==0.19.12

python-polars

1 respostas

  1. Best Answer

    Você precisa especificar o dtype interno:

    
In [59]: import polars as pl
    ...:
    ...: data = {"a": [ ["X"], ["Y"], []], "b": [3, 4, 5]}
    ...:
    ...: # this works normaly
    ...: df = pl.DataFrame(data)
    ...:
    ...: df_schema = [("a", pl.List(pl.Utf8)),
    ...:              ("b", pl.Int8)]
    ...:
    ...: df = pl.DataFrame(data, schema=df_schema)

In [60]: df
Out[60]:
shape: (3, 2)
┌───────────┬─────┐
│ a         ┆ b   │
│ ---       ┆ --- │
│ list[str] ┆ i8  │
╞═══════════╪═════╡
│ ["X"]     ┆ 3   │
│ ["Y"]     ┆ 4   │
│ []        ┆ 5   │
└───────────┴─────┘
    • 3

relate perguntas