从scala中的数据框添加两列

Question

我有两列年龄和工资存储在DF中。我只想编写一个scala代码来逐列添加这些值。我试过了

val age_1 = df.select("age")
val salary_1=df.select("salary")
val add = age_1+salary_1

给我错误。请帮忙

Answer 1

以下spark是SparkSession的实例，因此在实例化spark后，导入必须。

$ - 表示法可以通过导入带有

的spark implicits来使用

import spark.implicits._ 

然后使用$ -notation

val add = df.select($"age" + $"salary")

最终scala代码：

import spark.implicits._
val add = df.select($"age" + $"salary")

Apache doc