sum

Возвращает сумму всех значений в выражении.

Синтаксис

from pyspark.sql import functions as sf

sf.sum(col)

Параметры

Параметр	Тип	Description
col	pyspark.sql.Column или имя столбца	Целевой столбец для вычисления.

Возвраты

pyspark.sql.Column: столбец для вычисляемых результатов.

Примеры

Пример 1. Вычисление суммы значений в столбце

from pyspark.sql import functions as sf
df = spark.range(10)
df.select(sf.sum(df["id"])).show()

+-------+
|sum(id)|
+-------+
|     45|
+-------+

Пример 2. Использование выражения плюса для вычисления суммы

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

+------------+
|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.sum("age")).show()

+--------+
|sum(age)|
+--------+
|       6|
+--------+