主页 / coding / 问题 / 77441311
Accepted
datahack
Asked: 2023-11-08 04:36:28 +0800 CST

Polars 模式与列表类型不同

  • 772

我尝试创建带有两列的简单极坐标数据框：

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)

在不指定架构的情况下，它会创建以下数据框形状：

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

但是当我指定完全相同的模式时，它就会崩溃。可能是什么问题呢？

使用极坐标==0.19.12

python-polars

1 个回答

  1. Best Answer

    您需要指定内部数据类型：

    
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

相关问题