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

5 votes276 views1 answer
Torch concat matrix
Python
3 votes276 views1 answer
How to remove quotes from a string in python
Python
0 vote263 views1 answer
Get Project Parameter Dynamo Revit
Python
11 votes279 views3 answers
How to get the length of a string in python
Python
4 votes272 views1 answer
How to save matplotlib figure to png
Python
0 vote307 views1 answer
Flask print request headers
Python
1 vote291 views1 answer
Read a function of excel in python
Python
0 vote307 views1 answer
Python function guts
Python
1 vote294 views4 answers
Slice notation python
Python
3 votes269 views2 answers
Find different values from two lists python
Python