[英]Dask Dataframe split column of list into multiple columns
使用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
做类似的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'])})
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.