Skip to content

sum(col)

Описание

Функция sum() вычисляет сумму значений в группе. Это агрегатная функция, которая должна использоваться с groupBy().

Параметры

  • col: Column - числовой столбец для вычисления суммы

Возвращаемое значение

Тип значения из указанного столбца - сумма значений в группе

Пример использования

from pyspark.sql.functions import sum
from pyspark.sql import SparkSession

# Создаем SparkSession
spark = SparkSession.builder.appName("sum_example").getOrCreate()

# Создаем DataFrame
data = [
    ("A", 1),
    ("A", 2),
    ("A", 3),
    ("B", 4),
    ("B", 5),
    ("B", 6)
]
df = spark.createDataFrame(data, ["group", "value"])

# Вычисляем сумму для каждой группы
result = df.groupBy("group").agg(
    sum("value").alias("total")
).show()

# Результат:
# +-----+-----+
# |group|total|
# +-----+-----+
# |A    |6    |
# |B    |15   |
# +-----+-----+

Примечания

  • NULL значения игнорируются при вычислении суммы
  • Для вычисления среднего значения используйте avg()
  • Для подсчета количества значений используйте count()
  • Для вычисления произведения значений используйте product()