Dask Dataframe将列表的列拆分为多列
问题内容:
使用Pandas可以轻松完成相同的任务
import pandas as pd
df = pd.DataFrame({"lists":[[i, i+1] for i in range(10)]})
df[['left','right']] = pd.DataFrame([x for x in df.lists])
但是我不知道如何用 dask.dataframe
更新资料
到目前为止,我已经找到了解决方法
ddf = dd.from_pandas(df, npartitions=2)
ddf["left"] = ddf.apply(lambda x: x["lists"][0], axis=1, meta=pd.Series())
ddf["right"] = ddf.apply(lambda x: x["lists"][1], axis=1, meta=pd.Series())
我想知道是否还有另一种前进的方式。
问题答案:
您可以使用assign
以下方法实现此目的:
ddf = ddf.assign(left=ddf.lists.map(lambda x: x[0]),
right=ddf.lists.map(lambda x: x[1]))
例如,
ddf.compute()
lists left right
0 [0, 1] 0 1
1 [1, 2] 1 2
2 [2, 3] 2 3
3 [3, 4] 3 4
4 [4, 5] 4 5
5 [5, 6] 5 6
6 [6, 7] 6 7
7 [7, 8] 7 8
8 [8, 9] 8 9
9 [9, 10] 9 10
措辞的另一种方式(请参见下面的评论)可能是
ddf = ddf.assign(**{k: ddf.lists.map(lambda x, i=i: x[i])
for i, k in enumerate(['left', 'right'])})