try_sum

Возвращает сумму, вычисляемую из значений группы, и результатом является значение NULL при переполнении.

Синтаксис

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|
+------------+