Issue while creating new column in pyspark

Question

I have a python function which takes 2-input parameters and do some calculation and return some value

def func(column1,column2):
  if float(column1)!=1 and float(