我有一个小的 PySpark DataFrame df
:
index col1
0 1
1 3
2 4
还有一个字典:
LOOKUP = {0: 2, 1: 5, 2: 5, 3: 4, 4: 6}
我现在想要给
df
添加一个额外的列col2
,使其等于col1
的LOOKUP
值。
我的输出应该是这样的:
index col1 col2
0 1 5
1 3 4
2 4 6
我尝试使用:
df = df.withColumn(col("col2"), LOOKUP[col("col1")])
但是使用expr
和之前的方法都会报错。
在PySpark中如何实现这个?