小编Jus*_*vis的帖子

您可以将带有索引值的 pyspark 列传递给 element_at 函数吗？

我有一个类似于以下内容的 df：

------------------
 foo   |   bar
------------------
  1    |   [3, 2, 1]
  2    |   [3, 3, 1]
  1    |   [8, 2, 1]

我正在尝试使用 foo 作为索引来创建以下内容：

 baz  |
--------
  3   |
  3   |
  8   |

我正在尝试使用以下内容但没有成功：

df.withColumn('baz', F.element_at('bar', F.col('foo'))