mean(col)

Описание

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

Параметры

• col: Column - числовой столбец для вычисления среднего значения

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

Double - среднее арифметическое значение

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

from pyspark.sql.functions import mean
from pyspark.sql import SparkSession

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

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

# Вычисляем среднее значение
result = df.groupBy("group").agg(
    mean("value").alias("mean_value")
).show()

# Результат:
# +-----+----------+
# |group|mean_value|
# +-----+----------+
# |A    |2.0       |
# |B    |5.0       |
# +-----+----------+

Примечания

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