如何从列表中重命名 DataFrame 的列

Question

我有一个这样的数据框：

dog	cat
Cell 1	Cell 2
Cell 3	Cell 4

还有这样的列表：

dog, bulldog

cat, persian

我想创建一个函数来查找列表中列的名称并将其替换为第二个元素 (bulldog, persian)。

所以最后的结果应该是：

| bulldog  | persian  |
| -------- | -------- |
| Cell 1   | Cell 2   |
| Cell 3   | Cell 4   |

Answer 1

您需要为显示的 pre-defined 列表中的原始列执行 look-up。从中创建 Map 更容易，因此可以执行查找：

val list: List[(String, String)] = List(("dog", "bulldog"), ("cat", "persian"))

val columnMap = list.toMap

// columnMap: scala.collection.immutable.Map[String,String] = Map(dog -> bulldog, cat -> persian)


val originalCols = df.columns

val renamedCols = originalCols.map{
  c => if (columnMap.keys.toArray.contains(c)) s"${c} as ${columnMap.getOrElse(c, "")}"
       else c
}

println(renamedCols)

// renamedCols: Array[String] = Array(dog as bulldog, cat as persian)

df.selectExpr(renamedCols: _*).show(false)

// +-------+-------+
// |bulldog|persian|
// +-------+-------+
// |Cell 1 |Cell 2 |
// |Cell 1 |Cell 2 |
// +-------+-------+

如何从列表中重命名 DataFrame 的列

How to rename column of a DataFrame from a list

scala

apache-spark