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

PySpark DataFrame 的 head(~) 方法將前 n 行數作為 Row 對象返回。

參數

1.n | int | optional

要返回的行數。默認情況下，n=1 。

返回值

如果 n 大於 1，則返回 Row 對象列表
如果 n 等於 1，則返回單個 Row 對象 ( pyspark.sql.types.Row )

例子

考慮以下PySpark DataFrame：

columns = ["name", "age"]
data = [("Alex", 15), ("Bob", 20), ("Cathy", 25)]
df = spark.createDataFrame(data, columns)
df.show()



+-----+---+
| name|age|
+-----+---+
| Alex| 15|
|  Bob| 20|
|Cathy| 25|
+-----+---+

獲取 PySpark DataFrame 的第一行作為 Row 對象

要將第一行作為 Row 對象獲取：

df.head()



Row(name='Alex', age=15)

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

獲取前兩行作為 Row 對象的列表：

df.head(n=2)



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

相關用法

注：本文由純淨天空篩選整理自Isshin Inada大神的英文原創作品 PySpark DataFrame | head method。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。