gpt4 book ai didi

python - Spark DataFrame mapPartitions

转载 作者:太空狗 更新时间:2023-10-29 18:31:18 32 4
gpt4 key购买 nike

我需要在 Spark DataFrame 上进行分布式计算,在 DataFrame block 上调用一些任意(非 SQL)逻辑。我做了:

def some_func(df_chunk):
pan_df = df_chunk.toPandas()
#whatever logic here

df = sqlContext.read.parquet(...)
result = df.mapPartitions(some_func)

不幸的是,它会导致:

AttributeError: 'itertools.chain' object has no attribute 'toPandas'

我希望在每次映射调用中都有 spark DataFrame 对象,但我得到的是“itertools.chain”。为什么?以及如何克服这个问题?

最佳答案

试试这个:

>>> columns = df.columns
>>> df.rdd.mapPartitions(lambda iter: [pd.DataFrame(list(iter), columns=columns)])

关于python - Spark DataFrame mapPartitions,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/38749179/

32 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com