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函数中包含连续调用.