通过


try_sum

返回从组的值计算得出的总和,结果在溢出时为 null。

Syntax

from pyspark.sql import functions as sf

sf.try_sum(col)

参数

参数 类型 Description
col pyspark.sql.Column 或列名 要计算的目标列。

例子

示例 1:计算列中值的总和

from pyspark.sql import functions as sf
spark.range(10).select(sf.try_sum("id")).show()
+-----------+
|try_sum(id)|
+-----------+
|         45|
+-----------+

示例 2:结合使用加号表达式来计算总和

from pyspark.sql import functions as sf
df = spark.createDataFrame([(1, 2), (3, 4)], ["A", "B"])
df.select(sf.try_sum(sf.col("A") + sf.col("B"))).show()
+----------------+
|try_sum((A + B))|
+----------------+
|              10|
+----------------+

示例 3:使用 None 计算年龄的总和

import pyspark.sql.functions as sf
df = spark.createDataFrame([(1982, None), (1990, 2), (2000, 4)], ["birth", "age"])
df.select(sf.try_sum("age")).show()
+------------+
|try_sum(age)|
+------------+
|           6|
+------------+