Scala DataFrame单列重命名：如何将列key改为key2？

嘿，这事儿其实挺简单的，Scala对应的Spark API里专门有个方法能搞定单个列的重命名需求，把key改成key2直接用withColumnRenamed就好，给你一步步拆解：

示例代码 #

import org.apache.spark.sql.SparkSession

object RenameSingleColumnDemo {
  def main(args: Array[String]): Unit = {
    // 初始化SparkSession（本地调试用local模式，生产环境可以去掉master配置）
    val spark = SparkSession.builder()
      .appName("RenameColumnTest")
      .master("local[*]")
      .getOrCreate()
    
    import spark.implicits._
    
    // 先创建一个包含key列的示例DataFrame
    val originalDF = Seq(("apple", 5), ("banana", 3), ("orange", 7)).toDF("key", "count")
    
    // 核心操作：将key列重命名为key2
    val renamedDF = originalDF.withColumnRenamed("key", "key2")
    
    // 打印结果看看效果
    renamedDF.show()
  }
}

输出结果 #

+-------+-----+
|   key2|count|
+-------+-----+
|  apple|    5|
| banana|    3|
| orange|    7|
+-------+-----+

额外小提醒 #

如果怕列名拼写错误或者目标列不存在导致报错，可以先做个存在性检查：

val targetColumnName = "key"
if (originalDF.columns.contains(targetColumnName)) {
  val safeRenamedDF = originalDF.withColumnRenamed(targetColumnName, "key2")
  safeRenamedDF.show()
} else {
  println(s"警告：DataFrame中不存在列 $targetColumnName")
}

内容的提问来源于stack exchange，提问作者lak