Как получить среднее значение логического столбца в PySpark?

Question

Я пытался использовать .agg(avg("boolean_column")), но получил ошибку:

"function average requires numeric types, not boolean"

Как получить среднее значение для такого столбца?

Answer 1

Преобразовать столбец в числовой тип, затем взять среднее значение:

from pyspark.sql.functions import avg, col
df.groupBy(...).agg(avg(col("boolean_column").cast("double")))