You can use a "when otherwise" and give the condition you want. It works similar to sql case when query.
E.g.
df.select(when(df['col_1'] == 'A', "Condition1").otherwise("Condition2")).show()
You can use a "when otherwise" and give the condition you want. It works similar to sql case when query.
E.g.
df.select(when(df['col_1'] == 'A', "Condition1").otherwise("Condition2")).show()