Python PySpark DataFrame limit方法用法及代码示例

PySpark DataFrame 的 limit(~) 方法返回一个新的 DataFrame，其中包含指定的行数。

参数

1. num | number

返回所需的行数。

返回值

PySpark 数据帧 (pyspark.sql.dataframe.DataFrame)。

例子

考虑以下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|
+-----+-----+

使用 limit 获取 PySpark DataFrame 的行子集

要限制返回到 2 的行数：

df.limit(2).show()



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

请注意， show(~) 方法实际上有一个限制打印行数的参数：

df.show(n=2)



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

相关用法

注：本文由纯净天空筛选整理自Isshin Inada大神的英文原创作品 PySpark DataFrame | limit method。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。