scala - Spark数据框为每个现有行添加一行

Question

我有一个包含以下列的数据框:

groupid,unit,height
----------------------
1,in,55
2,in,54

我想用单位= cm和height = height * 2.54的其他行创建另一个数据框。

生成的数据帧:

groupid,unit,height
----------------------
1,in,55
2,in,54
1,cm,139.7
2,cm,137.16

不知道如何使用spark udf并在此处爆炸。
任何帮助表示赞赏。
提前致谢。

Answer 1

您可以使用dataframe创建另一个需要更改的withColumn，然后union将dataframes都设置为

import sqlContext.implicits._
import org.apache.spark.sql.functions._

val df = Seq(
  (1, "in", 55),
  (2, "in", 54)
).toDF("groupid", "unit", "height")

val df2 = df.withColumn("unit", lit("cm")).withColumn("height", col("height")*2.54)

df.union(df2).show(false)

你应该有

+-------+----+------+
|groupid|unit|height|
+-------+----+------+
|1      |in  |55.0  |
|2      |in  |54.0  |
|1      |cm  |139.7 |
|2      |cm  |137.16|
+-------+----+------+