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。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。