如何在检查 Spark scala 数据帧中的列是否存在后对其进行重命名？

Question

我需要重命名一个列，"DOB"在我的spark数据框架中添加"DateOfBirth"在检查该列在数据框中是否可用之后。

我正在做下面的重命名列，但是我如何在这里插入条件

df.withColumnRenamed("DOB","DateOfBirth")

有谁能帮帮忙吗?

Answer 1

您可以检查该列是否与column()函数一起存在，并且只在这种情况下重命名该列:

val renamed_df = 
if (df.columns.contains("DOB"))
df.withColumnRenamed("DOB","DateOfBirth")
else df