Python PySpark SQL Functions first方法用法及代碼示例

例子

考慮以下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 中列的第一個值

要獲取 name 列的第一個值：

import pyspark.sql.functions as F
df.select(F.first(df.name)).show()



+-----------+
|first(name)|
+-----------+
|       Alex|
+-----------+

獲取 PySpark DataFrame 中列的第一個非空值

考慮以下帶有空值的 PySpark DataFrame：

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



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

默認為ignorenulls=False，這意味著無論是否為null，都返回第一個值：

df.select(F.first(df.age)).show()



+----------+
|first(age)|
+----------+
|      null|
+----------+

要返回第一個非空值：

df.select(F.first(df.age, ignorenulls=True)).show()



+----------+
|first(age)|
+----------+
|        20|
+----------+

獲取PySpark中每組的第一個值

first(~) 方法在聚合中也很有用。考慮以下PySpark DataFrame：

columns = ["name", "class"]
data = [("Alex", "A"), ("Alex", "B"), ("Bob", None), ("Bob", "A"), ("Cathy", "C")]
df = spark.createDataFrame(data, columns)
df.show()



+-----+-----+
| name|class|
+-----+-----+
| Alex|    A|
| Alex|    B|
|  Bob| null|
|  Bob|    A|
|Cathy|    C|
+-----+-----+

要獲取每個聚合的第一個值：

df.groupby("name").agg(F.first("class")).show()



+-----+------------+
| name|first(class)|
+-----+------------+
| Alex|           A|
|  Bob|        null|
|Cathy|           C|
+-----+------------+

在這裏，我們按 name 進行分組，然後對於每個組，我們獲取 class 列中出現的第一個值。