使用输入中的列表动态定义数据框的名称和内容

Question

我编写了以下函数：

 def extractFromInput(inputName: String, colNames: List[String]): DataFrame = {

      val inputData: List[String] = inputName.split("-").map(_.trim).toList
      inputData.foldLeft(df) {
        (df: DataFrame, columnName: String) =>
          df.withColumn(columnName, lit(columnName))

      }
    }

我宁愿遍历colNames列表，并使用它来定义带有此列表内容的新列名称。 使用实际功能，各列的内容可以正确填充，但是我无法使它们正确命名。 知道如何执行此操作