variance(col)
Описание
Функция variance()
вычисляет выборочную дисперсию для числового столбца. Это агрегатная функция, которая должна использоваться с groupBy()
.
Параметры
col
: Column - числовой столбец для вычисления дисперсии
Возвращаемое значение
Double - выборочная дисперсия
Пример использования
from pyspark.sql.functions import variance
from pyspark.sql import SparkSession
# Создаем SparkSession
spark = SparkSession.builder.appName("variance_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(
variance("value").alias("variance")
).show()
# Результат:
# +-----+--------+
# |group|variance|
# +-----+--------+
# |A |1.0 |
# |B |1.0 |
# +-----+--------+
Примечания
- NULL значения игнорируются при вычислении дисперсии
- Для вычисления стандартного отклонения используйте
stddev_samp()
- Для вычисления генеральной дисперсии используйте
var_pop()
- Для вычисления среднего значения используйте
avg()