scala - 如何将标题和列添加到数据框 Spark ？

Question

我有一个数据框，我想在其上添加标题和第一列手动。这是数据框:

import org.apache.spark.sql.SparkSession 

val spark = SparkSession.builder.master("local").appName("my-spark-app").getOrCreate()
val df = spark.read.option("header",true).option("inferSchema",true).csv("C:\\gg.csv").cache()

数据框的内容

12,13,14
11,10,5
3,2,45

预期的输出是

define,col1,col2,col3
c1,12,13,14
c2,11,10,5
c3,3,2,45

Answer 1

你要做的是:

df.withColumn("columnName", column) //here "columnName" should be "define" for you

现在您只需要创建上述列(this 可能有帮助)