我正在pyspark中寻找与以下scala行等效的代码：

.map(i => get_json_object($"Payment", s"$$[$i]"))

尤其是 $$[$i] 它从json对象列表中获取第四个索引。
我基本上在这里寻找答案：dataframesparkscala分解json数组，但在pyspark中
谢谢。