加入df1和df2后，可以使用when和otherwise。为了使事情更具有一般性，我们可以编写一个函数，为给定的列创建when + otherwise语句。

def create_condition(col_name): 
    return F.when(
        F.col(col_name) == 'data1', F.col('data1')
    ).when(
        F.col(col_name) == 'data2', F.col('data2')
    ).otherwise(
        None
    )

df1.join(
    df2, on='common_column', how='inner'
).withColumn(
    "field1",
    create_condition("field1")
).withColumn(
    "field2",
    create_condition("field2")
)

+-------------+-----+-----+------+------+
|common_column|data1|data2|field1|field2|
+-------------+-----+-----+------+------+
|            A|    1|    2|     1|     2|
|            B|    3|    4|     4|     3|
|            C|    5|    6|     5|  null|
+-------------+-----+-----+------+------+