Python PySpark DataFrame take方法用法及代碼示例

例子

考慮以下PySpark DataFrame：

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



+-----+---+
| name|age|
+-----+---+
| Alex| 25|
|  Bob| 30|
|Cathy| 40|
+-----+---+

獲取 PySpark DataFrame 的前 n 行作為 Row 對象列表

獲取前 n 行數作為 Row 對象列表：

df.take(2)



[Row(name='Alex', age=25), Row(name='Bob', age=30)]

take(~) 和 head(~) 方法之間的區別

方法 takes(~) 和 head(~) 之間的區別是始終返回 Row 對象列表，而 head(~) 在設置 head(n=1) 的情況下僅返回 Row 對象。

例如，考慮以下PySpark DataFrame：

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



+----+---+
|name|age|
+----+---+
|Alex| 20|
| Bob| 30|
+----+---+

調用 take(1) 會產生：

df.take(1)



[Row(name='Alex', age=20)]

調用 head(1) 會產生：

df.head(1)



[Row(name='Alex', age=20)]

對於 n 的所有其他值，方法 take(~) 和 head(~) 產生相同的輸出。