/Python/Create spark dataframe from pandas

Create spark dataframe from pandas

Create spark dataframe in python

spark.createDataFrame(["10","11","13"], "string").toDF("age")

Source: stackoverflow.com

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

0 vote256 views1 answer
Torch.unsqueze
Python
0 vote211 views1 answer
Dict column to be in multiple columns python
Python
0 vote188 views1 answer
Python make an object hashable
Python
1 vote193 views1 answer
Python array spread
Python
0 vote187 views1 answer
Django multi column index
Python
0 vote204 views1 answer
Shorten all floats in a list
Python
46 votes189 views6 answers
Dataframe to CSV
Python
0 vote189 views1 answer
Python select columns with no na
Python
50 votes189 views6 answers
Print random string from list python
Python
0 vote203 views1 answer
Convert math expression as string to int
Python