如何将字典键值项目应用于 spark 中数据集中的列？

Question

spark的新成员...如何使用spark数据集中的列ask键获取一些值并将这些值作为新列添加到数据集中？

在python中，我们有类似的东西：

 df.loc[:,'values'] = df.loc[:,'key'].apply(lambda x: D.get(x))

其中D是前面定义的python中的函数。

如何使用Java在spark中实现这一点？非常感谢。

编辑：例如：我有以下数据集df：

A
1
3
6
0
8

我想根据以下字典创建一个工作日列：

D[1] = "Monday"
D[2] = "Tuesday"
D[3] = "Wednesday"
D[4] = "Thursday"
D[5] = "Friday"
D[6] = "Saturday"
D[7] = "Sunday"

并将列添加回我的数据集df：

A    days
1    Monday
3    Wednesday
6    Saturday
0    Sunday
8    NULL

这只是一个示例，列A当然可以是整数以外的任何东西。

Answer 1

1. 您可以使用df.withColumn返回一个新的df，其中包含新的列值和df的前一个值。
2. 创建一个udf函数（用户定义函数）来应用字典映射。

下面是一个可重现的示例：

>>> from pyspark.sql.types import StringType 
>>> from pyspark.sql.functions import udf 
>>> df = spark.createDataFrame([{'A':1,'B':5},{'A':5,'B':2},{'A':1,'B':3},{'A':5,'B':4}], ['A','B'])
>>> df.show() 
+---+---+
|  A|  B|
+---+---+
|  1|  5|
|  5|  2|
|  1|  3|
|  5|  4|
+---+---+

>>> d = {1:'x', 2:'y', 3:'w', 4:'t', 5:'z'}
>>> mapping_func = lambda x: d.get(x) 
>>> df = df.withColumn('values',udf(mapping_func, StringType())("A"))
>>> df.show() 
+---+---+------+
|  A|  B|values|
+---+---+------+
|  1|  5|     x|
|  5|  2|     z|
|  1|  3|     x|
|  5|  4|     z|
+---+---+------+