pysparkで、各レコードが保持する２つの値に対して大きい方の値を取得して新規カラムとして保持する方法

Question

標題の通り、pyapark Dataframeについて、２つの列の値を比較して新しい列を作成する方法をご教授いただきたいです。

s = """a,b
10,1
9,2 
8,3
7,4
6,5
5,6
4,7
3,8
2,9
1,10"""
df = spark.read.csv(StringIO(s), parse_dates=['entry_time'])
＃　各業に対してdf[a], df[b]のうち大きい方の値を採用し、df[c]としてカラム追加

psandas Dataframeであれば

iimport numpy as np

df['C'] = np.where(df['A']>df['B'], df['A'], df['B'])

という形で実現可能かと思いますが、pysparkで同様の操作を行うためのメソッド等御座いますでしょうか？
どうぞよろしくお願いいたします。

Answer 1

下記記事が参考になるかと思います。

Pyspark dataframe操作

列の追加

case whenの様に列を追加する
.withColumn('new_column_name', when(col('colname') === "hoge", 1).otherwise(0))

ドキュメントは下記です。
DataFrame.withColumn(colName, col)

Column.when(condition, value)