Scala中DataFrame指定列批量重命名失效，求正确实现方案

原代码的核心问题在于：

• DataFrame是不可变对象，每次调用withColumnRenamed都会生成新的DataFrame实例，而非修改原对象
• 循环中未将重命名后的新DataFrame赋值给变量，所有操作结果都被丢弃，最终返回的仍是原始DataFrame，仅最后一列看似生效是巧合（实际未正确累积修改）

方案一：使用foldLeft累积修改（适合少量列） #

利用foldLeft从原始DataFrame开始，依次对指定列执行重命名，逐步累积修改结果：

def prefixColumns(dataframe: DataFrame, columnPrefix: String, cols: Seq[String]): DataFrame = {
  cols.foldLeft(dataframe) { (currentDf, colName) =>
    currentDf.withColumnRenamed(colName, s"${columnPrefix}_$colName")
  }
}

方案二：使用select构造列表达式（性能更优，适合大量列） #

直接遍历所有列，构造重命名后的列表达式，一次生成最终DataFrame：

def prefixColumns(dataframe: DataFrame, columnPrefix: String, cols: Seq[String]): DataFrame = {
  val columnExprs = dataframe.columns.map { colName =>
    if (cols.contains(colName)) {
      col(colName).as(s"${columnPrefix}_$colName")
    } else {
      col(colName)
    }
  }
  dataframe.select(columnExprs: _*)
}

保持原有调用逻辑即可（注意列名需为字符串类型）：

prefixColumns(products, "products", Seq("col1", "col2", "col3", "col4"))

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