scala - 创建一个 bool 特征来检查两列是否相同

Question

我有一个数据框 DF1，它具有三个特征(列)a、b、c，均为 StringType。我想从 DF1 创建一个新的数据框 DF2，它有两列:

1. a列
2. 如果 b=c 则为 1 的新列 d，否则为 0

输入示例:

a b c  
A B B  
B C A  
D D D  

想要的输出

a d  
A 1  
B 0  
D 1  

Answer 1

缺少的部分是其他两列的 drop。

val df2 = df1.withColumn("d", col("b") === col("c")).drop("b").drop("c")
df2.show

这给了我们

+---+-----+
|  a|    d|
+---+-----+
|  A| true|
|  B|false|
|  D| true|
+---+-----+