pyspark when condition

Busy Buffalo answered on January 26, 2023 Popularity 9/10 Helpfulness 3/10

answer pyspark when condition

related pyspark when

pyspark when condition

Comment

df = df.withColumn("outcome", F.when(F.col("v1") == 0, 1)
      .when(F.col("v1") >0, 2)
      .otherwise(0))

Popularity 9/10 Helpfulness 3/10 Language whatever

Source: stackoverflow.com

Tags: pyspark whatever when

Link to this answer
Share Copy Link

Contributed on Jan 26 2023

Busy Buffalo

0 Answers Avg Quality 2/10

Closely Related Answers

pyspark when

Comment

from pyspark.sql.functions import when
df.select("*",when(df.value == 1, 'one').when(df.value == 2, 'two').otherwise('other').alias('value_desc')).show()
# Alternative
from pyspark.sql.functions import expr 
df.withColumn("value_desc",expr("CASE WHEN value == 1 THEN  'one' WHEN value == 2 THEN  'two' ELSE 'other' END AS value_desc")).show()

Popularity 6/10 Helpfulness 6/10 Language python

Source: stackoverflow.com

Tags: pyspark python when

Link to this answer
Share Copy Link