Get first element from PySpark data frame

user_5451971 0
Viewed 46

I have a PySpark data frame which only contains one element. How can I extract the number from the data frame?

For the example, how can I get the number 5.0 from the PySpark data frame?

+-----------------+
|        count    |
+-----------------+
|        5.0      |
+-----------------+
pyspark apache-spark dataframe extract
2 Answers

Several alternatives:

df.head()[0]
df.head().count
df.head()['count']

first does the same, but head is 1 letter shorter ;)

user_27535010 
sc.parallelize([5.0]).collect()

[5.0]
user_195068940
Related