如何使用 Palantir Foundry 在 Pyspark 中編寫 case 語句

Question

試圖從 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

Answer 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)
)