需求：将Pandas中Lambda实现的default_flag1列逻辑转换为PySpark代码

原Pandas代码的逻辑是根据三个条件判断生成default_flag1列：当A等于1、crr属于('sss','tttt')、reg等于'T'这三个条件满足任意一个时，值为'T'，否则为'F'。对应的PySpark实现如下：

from pyspark.sql.functions import when, col

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

说明 #

• 用withColumn添加新列，相比Pandas的行遍历apply，更适配PySpark的分布式计算模型，性能更优
• 通过col()引用DataFrame列，用|表示逻辑或，isin()实现集合成员判断
• when-otherwise结构对应原代码的三元表达式逻辑

内容的提问来源于stack exchange，提问作者Hala El Henawy