def sample(a:Column,b:Column):Column={ var c:Column=null ;c=a*(1-b);c} error: overloaded method value + with alternatives cannot be applied to (org.apache.spark.sql.Column)

[#1]

Hi! I'm trying to define a Scala function in Apache Spark that performs some column arithmetic, but I am running into a compilation error.

Here is my code:

Scala

def sample(a: Column, b: Column): Column = { 
  var c: Column = null
  c = a * (1 - b)
  c
}

When I try to use this, I get the following error message:

error: overloaded method value + with alternatives cannot be applied to (org.apache.spark.sql.Column)

How can I fix this arithmetic expression so it works with Spark Column objects?

[#2 · Accepted solution]

c=a*(lit(1)-b)