CODEWITHSUNDEEP
pythonapache-sparkpysparkdatabricks
STORM
STORM

Reputation: 4331

Cannot create Dataframe in PySpark

I want to create a Dataframe in PySpark with the following code

from pyspark.sql import *
from pyspark.sql.types import *

temp = Row("DESC", "ID")
temp1 = temp('Description1323', 123)

print temp1

schema = StructType([StructField("DESC", StringType(), False),
                     StructField("ID", IntegerType(), False)])

df = spark.createDataFrame(temp1, schema)

But i am receiving the following error:

TypeError: StructType can not accept object 'Description1323' in type type 'str'

Whats wrong with my code?

Upvotes: 2

Views: 4224

Answers (1)

martinarroyo
martinarroyo

Reputation: 9701

The problem is that you are passing a Row where you should be passing a list of Rows. Try this:

from pyspark.sql import *
from pyspark.sql.types import *

temp = Row("DESC", "ID")
temp1 = temp('Description1323', 123)

print temp1

schema = StructType([StructField("DESC", StringType(), False),
                     StructField("ID", IntegerType(), False)])

df = spark.createDataFrame([temp1], schema)

df.show()

And the result:

+---------------+---+
|           DESC| ID|
+---------------+---+
|Description1323|123|
+---------------+---+

Upvotes: 4

Related Questions