spark-koalas - Spark-Koalas 错误：列分配不支持类型元组

Question

我无法分配kdf[c].factorize()给kdf[c].

我试过这个，但没有帮助：

ks.set_option('compute.ops_on_diff_frames', True)
kdf[response] = ks.Series(kdf[response].factorize())
ks.reset_option('compute.ops_on_diff_frames')

任何帮助表示赞赏。

Answer 1

factorize返回一个元组(Series, Index)。您可以将元组的第一个元素分配给响应列。它已经是一个系列，因此无需使用ks.Series. 试试这个代码：

kdf[response] = kdf[response].factorize()[0]

PS 无需设置ops_on_diff_frames，因为您在这里只处理单个数据帧。