power(col1, col2)
Описание
Функция power()
возводит число в указанную степень.
Параметры
col1
: Column - основание степени (число, которое нужно возвести в степень)col2
: Column - показатель степени
Возвращаемое значение
Column - результат возведения в степень
Пример использования
from pyspark.sql.functions import power, lit
from pyspark.sql import SparkSession
# Создаем SparkSession
spark = SparkSession.builder.appName("power_example").getOrCreate()
# Создаем DataFrame с числами
data = [
(2, 3),
(3, 2),
(4, 0.5)
]
df = spark.createDataFrame(data, ["base", "exponent"])
# Вычисляем степень
df = df.withColumn("result", power("base", "exponent"))
df.show(truncate=False)
# Результат:
# +----+--------+------+
# |base|exponent|result|
# +----+--------+------+
# |2 |3 |8.0 |
# |3 |2 |9.0 |
# |4 |0.5 |2.0 |
# +----+--------+------+
Примечания
- Возвращает NULL, если любой из аргументов NULL
- Поддерживает дробные степени (например, 0.5 для квадратного корня)
- Эквивалентна оператору
**
в Python - Для извлечения квадратного корня можно использовать
sqrt()
- Для логарифмов используйте:
log()
для натурального логарифмаlog10()
для десятичного логарифмаlog2()
для логарифма по основанию 2