PySpark：当具有多个输出的功能时[重复]

Question

我正在尝试使用“链接时”功能。 换句话说，我想得到两个以上的输出。

我尝试在 Excel 中使用与连接 IF 函数相同的逻辑：

  df.withColumn("device_id", when(col("device")=="desktop",1)).otherwise(when(col("device")=="mobile",2)).otherwise(null))

但这不起作用，因为我无法将元组放入“否则”函数中。

Answer 1

你试过了吗：

from pyspark.sql import functions as F
df.withColumn('device_id', F.when(col('device')=='desktop', 1).when(col('device')=='mobile', 2).otherwise(None))

请注意，在链接 when 函数时，您不需要将连续调用包装在 otherwise 函数中。