当条件不起作用时使用多个 Pyspark

Question

我的数据集看起来像这样

原始数据

我编写了以下代码：

flightData2015.select("*",when(flightData2015['count']>200,'above200')
                  .when(flightData2015['count']>400,'above400').otherwise("below").alias("new count")).show()

输出： 红线不符合我的逻辑，我想知道为什么第二个“when”条件不起作用

Answer 1

首先，条件“>200”也将满足大于400的项目，因此这就是不使用第二个when的原因。

其次，嵌套的when else 子句应该有效。

when(flightData2015['count']>400,'above400').otherwise(when(flightData2015['count']>200,'above200').otherwise("below"))