PySpark：替换基于准则

Question

我具有低于

+++++++++++++++++++++ 
colA | colB | colC | 
+++++++++++++++++++++ 
123 | 3 | 0| 
222 | 0 | 1| 
200 | 0 | 2| 

一个数据帧为数据帧显示的值I要为1的值来代替colB和colC的值如果它们是大于0

我能够使用na.fill函数，如果我需要填充0的零。但我不知道如何做到这一点。

Answer 1

假设你的数据帧为df，那么你就可以做到以下几点：

from pyspark.sql.functions import when 

df = df.select('colA', 
        when(df.colB > 0, 1).alias('colB'), 
        when(df.colB > 0, 1).alias('colC')) 

此检查COLB和COLC是否大于0和1分配