Pyspark when otherwise column value. The same can be implemented directly ...