Convert a spark DataFrame to pandas DF

Is there a way to convert a Spark Df (not RDD) to pandas DF

I tried the following:

var some_df = Seq( ("A", "no"), ("B", "yes"), ("B", "yes"), ("B", "no") ).toDF( "user_id", "phone_number") 

Code:

%pyspark pandas_df = some_df.toPandas() 

Error:

 NameError: name 'some_df' is not defined 

Any suggestions.

10

3 Answers

following should work

some_df = sc.parallelize([ ("A", "no"), ("B", "yes"), ("B", "yes"), ("B", "no")] ).toDF(["user_id", "phone_number"]) pandas_df = some_df.toPandas() 
4

In my case the following conversion from spark dataframe to pandas dataframe worked:

pandas_df = spark_df.select("*").toPandas() 
4

Converting spark data frame to pandas can take time if you have large data frame. So you can use something like below:

spark.conf.set("spark.sql.execution.arrow.enabled", "true") pd_df = df_spark.toPandas() 

I have tried this in DataBricks.

2

Your Answer

Sign up or log in

Sign up using Google Sign up using Facebook Sign up using Email and Password

Post as a guest

By clicking “Post Your Answer”, you agree to our terms of service, privacy policy and cookie policy

You Might Also Like