Python PySpark SQL Functions least方法用法及代码示例

例子

考虑以下PySpark DataFrame：

df = spark.createDataFrame([[20,10], [30,50], [40,90]], ["A", "B"])
df.show()



+---+---+
|  A|  B|
+---+---+
| 20| 10|
| 30| 50|
| 40| 90|
+---+---+

获取PySpark中指定列的每一行的最小值

要获取 A 和 B 列的每一行的最小值：

import pyspark.sql.functions as F
df.select(F.least("A","B")).show()



+-----------+
|least(A, B)|
+-----------+
|         10|
|         30|
|         40|
+-----------+

我们还可以传递 Column 对象而不是列标签：

df.select(F.least(df.A,df.B)).show()



+-----------+
|least(A, B)|
+-----------+
|         10|
|         30|
|         40|
+-----------+

请注意，我们可以使用 withColumn(~) 附加由 least(~) 返回的 Column ：

df.withColumn("smallest", F.least("A","B")).show()



+---+---+--------+
|  A|  B|smallest|
+---+---+--------+
| 20| 10|      10|
| 30| 50|      30|
| 40| 90|      40|
+---+---+--------+