Spark to pandas

Convert pandas dataframe to spark dataframe

import pandas as pd
from pyspark.sql import SparkSession

filename = <'path to file'>
spark = SparkSession.build.appName('pandasToSpark').getOrCreate()
# Assuming file is csv
pandas_df = pd.read_csv(filename)
spark_df = spark.CreateDataFrame(pandas_df)

Dataframe pandas to spark

from pyspark.sql import SparkSession
#Create PySpark SparkSession
spark = SparkSession.builder \
    .master("local[1]") \
    .appName("SparkByExamples.com") \
    .getOrCreate()
#Create PySpark DataFrame from Pandas
sparkDF=spark.createDataFrame(pandasDF) 
sparkDF.printSchema()
sparkDF.show()

#Outputs below schema & DataFrame

root
 |-- Name: string (nullable = true)
 |-- Age: long (nullable = true)

+------+---+
|  Name|Age|
+------+---+
| Scott| 50|
|  Jeff| 45|
|Thomas| 54|
|   Ann| 34|
+------+---+

Source: sparkbyexamples.com

Create spark dataframe from pandas

import numpy as np
import pandas as pd

# Enable Arrow-based columnar data transfers
spark.conf.set("spark.sql.execution.arrow.enabled", "true")

# Generate a pandas DataFrame
pdf = pd.DataFrame(np.random.rand(100, 3))

# Create a Spark DataFrame from a pandas DataFrame using Arrow
df = spark.createDataFrame(pdf)

Similar codes

2 votes255 views1 answer
Json dumps datetime
Python
0 vote200 views1 answer
Python create sqlite db in memory
Python
0 vote216 views1 answer
Performance of extend vs append loop
Python
0 vote216 views1 answer
Except Exception, e: suds python
Python
0 vote229 views1 answer
Windows how to store filepath as variabley python
Python
0 vote215 views1 answer
How do variables work in python
Python
5 votes213 views2 answers
Regular expression to remove space python
Python
1 vote205 views2 answers
How to print 0 to 10 in python
Python
11 votes204 views3 answers
How to strip white space of text in python?
Python
0 vote225 views1 answer
Django update request.post
Python