Если...

Оценка списка условий и возвращение одного из нескольких возможных выражений результатов. Если otherwise() не вызывается, значение None возвращается для несовпаденных условий. Поддерживает Spark Connect.

Синтаксис

from pyspark.databricks.sql import functions as dbf

dbf.when(condition=<condition>, value=<value>)

Параметры

Параметр	Тип	Description
condition	pyspark.sql.Column	Логическое выражение column.
value	Любое	Литеральное значение или выражение столбца.

Возвраты

pyspark.sql.Column: столбец, представляющий выражение.

Примеры

from pyspark.databricks.sql import functions as dbf
df = spark.range(3)
df.select("*", dbf.when(df['id'] == 2, 3).otherwise(4)).show()

+---+------------------------------------+
| id|CASE WHEN (id = 2) THEN 3 ELSE 4 END|
+---+------------------------------------+
|  0|                                   4|
|  1|                                   4|
|  2|                                   3|
+---+------------------------------------+