Python lambda to pyspark

I have this Python code written in pandas, I need to write the same in Pyspark:

Source_df_write['default_flag1']=Source_df_write.apply(lambda x: 'T' if ((x['A']==1) or (x['crr'] in ('sss','tttt')) or (x['reg']=='T')) else 'F', axis=1)

>Solution :

You can use when and otherwise:

import pyspark.sql.functions as F

Source_df_write.withColumn("default_flag1", F.when(
    (F.col("A") == 1) | (F.col("crr").isin(["sss","tttt"])) | (F.col("reg") == 'T'), "T"
).otherwise("F"))