python-3.x - 将 Pandas 数据帧转换为 PySpark 数据帧

Question

这个问题在这里已经有了答案:





Convert between spark.SQL DataFrame and pandas DataFrame [duplicate]

(1 个回答)


2年前关闭。




我有一个具有以下设置的脚本。

我在用:

1) Spark 数据帧来拉取数据
2)初始聚合后转换为pandas数据帧
3) 想转换回 Spark 写入 HDFS

从 Spark --> Pandas 的转换很简单，但我在如何将 Pandas 数据帧转换回 Spark 方面苦苦挣扎。

你能建议吗？

from pyspark.sql import SparkSession
import pyspark.sql.functions as sqlfunc
from pyspark.sql.types import *
import argparse, sys
from pyspark.sql import *
import pyspark.sql.functions as sqlfunc
import pandas as pd

def create_session(appname):
    spark_session = SparkSession\
        .builder\
        .appName(appname)\
        .master('yarn')\
        .config("hive.metastore.uris", "thrift://uds-far-mn1.dab.02.net:9083")\
        .enableHiveSupport()\
        .getOrCreate()
    return spark_session
### START MAIN ###
if __name__ == '__main__':
    spark_session = create_session('testing_files')

我已经尝试了以下 - 没有错误，只是没有数据!确认一下，df6 确实有数据 & 是一个 Pandas 数据框

df6 = df5.sort_values(['sdsf'], ascending=["true"])
sdf = spark_session.createDataFrame(df6)
sdf.show()

Answer 1

开始了:

# Spark to Pandas
df_pd = df.toPandas()

# Pandas to Spark
df_sp = spark_session.createDataFrame(df_pd)