如何使用 Palantir Foundry 在 Pyspark 中编写案例陈述 [重复]

【问题标题】：How do I write case statements in Pyspark using Palantir Foundry [duplicate]如何使用 Palantir Foundry 在 Pyspark 中编写案例陈述 [重复]
【发布时间】：2021-12-12 05:07:18
【问题描述】：

试图从 VBA 中的宏中获取 case 语句并转移到 Pyspark。我将如何在 Pyspark 中编写这样的内容？

 Select Case dMinutes
                Case 0 To 30
                    xnum = 1
                Case 31 To 60
                    xnum = 2
                Case 61 To 90
                    xnum = 3
                Case 91 To 120
                    xnum = 4
                Case Else
                    xnum = 5

【问题讨论】：

这能回答你的问题吗？ Apache spark dealing with case statements

标签： sql pyspark switch-statement case palantir-foundry

【解决方案1】：

下面的 PySpark 应该可以实现你想要的！

import pyspark.sql.functions as F

dataset = dataset.withColumn('xnum',
    F.when(F.col('dMinutes').between(0,30), 1)
    .when(F.col('dMinutes').between(31,60), 2)
    .when(F.col('dMinutes').between(61,90), 3)
    .when(F.col('dMinutes').between(91,120), 4)
    .otherwise(5)
)

【讨论】：