
        
        
            
                
                
                    df = spark.createDataFrame([["Alex", 25], ["Bob", 30], ["Cathy", 40]], ["name", "age"])
df.show()
                
            
            +-----+---+
| name|age|
+-----+---+
| Alex| 25|
|  Bob| 30|
|Cathy| 40|
+-----+---+

Getting the first n number of rows of PySpark DataFrame as list of Row objects

To get the first n number of rows as list of Row objects:


        
        
            
                
                
                    df.take(2)
                
            
            [Row(name='Alex', age=25), Row(name='Bob', age=30)]

Difference between methods take(~) and head(~)

The difference between methods takes(~) and head(~) is takes always return a list of Row objects, whereas head(~) will return just a Row object in the case when we set head(n=1).

For instance, consider the following PySpark DataFrame:


        
        
            
                
                
                    df = spark.createDataFrame([["Alex", 20], ["Bob", 30]], ["name", "age"])
df.show()
                
            
            +----+---+
|name|age|
+----+---+
|Alex| 20|
| Bob| 30|
+----+---+

Invoking take(1) yields:


        
        
            
                
                
                    df.take(1)
                
            
            [Row(name='Alex', age=20)]

Invoking head(1) yields:


        
        
            
                
                
                    df.head(1)
                
            
            [Row(name='Alex', age=20)]

For all other values of n, the methods take(~) and head(~) yield the same output.

Published by Isshin Inada

Edited by 0 others

Did you find this page useful?

thumb_up

thumb_down

Comment

Citation

Ask a question or leave a feedback...

Official PySpark Documentation

https://spark.apache.org/docs/3.1.1/api/python/reference/api/pyspark.sql.DataFrame.take.html

thumb_up

thumb_down

chat_bubble_outline

settings

Enjoy our search

Hit / to insta-search docs and recipes!