在SparkR中，我們如何在現有列上添加基於邏輯操作的新列？

Question

我可以很容易地做到這一點

employees$salarynew = employees$Salary * 2 
employees$sale_chance <- (employee$Salary > 2000) 

但是，如果我想是這樣的

employees$sale_chance_1 <- if (employees$Salary > 2000) 'high' else 'low' 

它給出錯誤

Error in if (employees$Salary > 2000) "high" else "low" : 
    argument is not interpretable as logical 

這也給了同樣的錯誤

emp1 <- withColumn(employees, "sale_prob", {if (employees$Salary > 2000) 'high' else 'low'}) 

Answer 1

它不應該是這樣的ifelse(employees$Salary > 2000, 'high', 'low')

Answer 2

正如評論

emp$sale = ifelse(emp$Salary > 2000, 'high', 'low') 

也有人，這適用於更復雜的表達式

emp1 <- selectExpr(emp, "case Salary>2000 when TRUE then 'high' else 'low' end as sale_prob")