首页 > 解决方案 > 来自python元组列表的dask数据框

问题描述

我对 dask 真的很陌生。我想从 python 元组列表创建一个 dask 数据框。在 pandas 中,您可以使用DataFrame.from_records将元组列表转换为数据框。什么功能可以在 dask 中为我提供相同的功能。我的数据看起来有点像这样

[(21262, 'booking', 'NULL'), (21262, 'booking', 'NULL'), (21262, 'booking', 'NULL'), (21262, 'booking', ''), (21262 , '预订', 'NULL')]

我现在正在使用此代码执行任务。这是这样做的正确方法吗。

import pandas as pd
import dask
import dask.dataframe as dd

names = ['id', 'status', 'reg_entry']
dfs = dask.delayed(pd.DataFrame.from_records)(cursor.fetchall(), columns=names)

df = dd.from_delayed(dfs)

标签: pythonpandasdataframedask

解决方案


您可以尝试从现有的 pandas 数据框创建一个 dask 数据框(以便能够使用所有 pandas 构造函数):

df = pd.DataFrame([(21262, 'booking', 'NULL'), (21262, 'booking', 'NULL'), (21262, 'booking', 'NULL'), (21262, 'booking', ''), (21262, 'booking', 'NULL')])
ddf = dd.from_pandas(df, npartitions=2)

推荐阅读