pysparkで新しい列を作成するには？

Question

私のpyspark DataFrameには、2つのカラムprice1とprice2があります。私は式((price1 - price2)/price1)に基づいて新しい列resultを作成したいと思います。しかし、price1でもprice2もヌルでないことを確認したいと思います。price1は0ではありません。

これらの条件を使用して新しい列を正しく作成するにはどうすればよいですか？あなたはUDFを使用することができる場合、私はあなたを考える

Answer 1

..

from pyspark.sql import functions as F 

udf = F.udf(lambda x,y : 0 if x == 0 or not all((x,y)) else x-y/x) 
df = df.withColumn("result", udf(df["price1"],df["price2"])) 

Answer 2

：

今、私はこれを持っていますこのようにすることができます：

df = df.withColumn("result", df["price1"]-df["price1"]/df["price2"]).fillna(0) 

Answer 3

df = df.withColumn("result", 
when(df.col("price1").isNull OR df.col("price2").isNull OR df.col("price1")==0,0) 
.otherwise(df.col("price1")-df.col("price2")/df.col("price1"))) 

これはそれがスカラ座を使用して行うことができる方法で、

df = df.withColumn("result", df["price1"]-df["price2"]/df["price1"])