スパーク：条件式と列操作

Question

Iは、以下のスキーマでデータフレームを有する：上記の例で

if cnt > 1: 
    colC = colA + colB 
else: 
    colC = colA 

：私は、以下の条件式（擬似コード）を実行する

|- colA (Int) 
|- colB (Int) 
|- cnt (Int) 

を、colCは、withColumn関数で作成された新しい列です。

collectをcntに送信したくありません。

Answer 1

あなたは条件付きで列を作成するためにwithColumnと組み合わせwhen/otherwise構文を使用することができます。

df.withColumn("colC", when($"cnt" > 1, $"colA" + $"colB").otherwise($"colA"))