使用数据构造表。避免创建数据框

时间:2019-06-17 21:24:53

标签: pyarrow

熊猫数据框很重,所以我想避免这种情况。 但是我想构造Pyarrow Table以便以拼花形式存储数据。

我搜索并阅读了文档,然后尝试使用     from_array() 但它不起作用。

field=[pa.field('name',pa.string()),pa.field('age',pa.int64())]
arrays=[pa.array(['Tom']),pa.array([23])]
pa.Table.from_arrays(pa.schema(field),arrays)

错误是:     名称(1)的长度与数组(2)的长度不匹配

1 个答案:

答案 0 :(得分:0)

在此处查看Table.from_arrays文档:https://arrow.apache.org/docs/python/generated/pyarrow.Table.html#pyarrow.Table.from_arrays 它期望的第一个参数是数组,而不是模式。因此,您可以执行以下操作:

In [64]: pa.Table.from_arrays(arrays, schema=pa.schema(field))
Out[64]: 
pyarrow.Table
name: string
age: int64

或传递列名而不是完整模式:

In [65]: pa.Table.from_arrays(arrays, names=['name', 'age']) 
Out[65]: 
pyarrow.Table
name: string
age: int64

在下一版本的pyarrow(0.14.0)中,您还可以执行以下操作:

In [51]: pa.Table.from_pydict({'name': pa.array(['Tom']), 'age': pa.array([23])})
Out[51]: 
pyarrow.Table
name: string
age: int64