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