如何计算DataFrame列的2的幂

Question

我需要p使用Spark 2.2和Scala 计算列的2的幂:

但如果我这样做,我得到错误,因为($"ki" / $"ni")是列,而不是Double.

df.withColumn("p",(lit(1) - scala.math.pow(($"ki" / $"ni").as[Double],2))

Answer 1

您可以使用内置的战俘功能为

import org.apache.spark.sql.functions._
df.withColumn("power_of_two", pow($"p", lit(2)))