在Spark数据框中划分2列的值

Question

我在Spark中将此表存储为Dataframe。我想得到另一个包含比率的数据帧（url，url1，ratio），其中比率= count1 / count。如何编写操作呢？

Answer 1

您可以使用以下内容：

import spark.implicits._
val newDF = df.withColumn("ratio", $"count1" / $"count") 

这行代码将添加一个名为列ration到你df和结果存储在newDF

编辑1 ：（按要求提供Java解决方案）

import org.apache.spark.sql.functions._
Dataset<Row> newDF = df.withColumn("ration", col("count1").divide(col("count")))