pyspark.sql.DataFrame.take#

DataFrame.take(num)[source]#

Returns the first num rows as a list of Row.

Added in version 1.3.0.

Changed in version 3.4.0: Supports Spark Connect.

Parameters:
numint

Number of records to return. Will return this number of records or all records if the DataFrame contains less than this number of records..

Returns:
list

List of rows

Examples

>>> df = spark.createDataFrame(
...     [(14, "Tom"), (23, "Alice"), (16, "Bob")], ["age", "name"])

Return the first 2 rows of the DataFrame.

>>> df.take(2)
[Row(age=14, name='Tom'), Row(age=23, name='Alice')]