从同一pyspark数据帧的键数组中获取值数组

我的pyspark数据框架如下所示。它有三列

col1:key
col2：值 col3：数组\u的\u键

我想添加一个新列col4，其中包含键的值数组。所以它应该像下面给出的第二幅图。有人能帮我找到一个有效的方法吗

现在