从数组 pySpark 内的元组中获取第 n 个元素

Question

我正在使用 pySpark，我想从一列的元组列表中检索一个值。 当我在使用 toPandas() 后打印它时，结构看起来像这样。

[[Row(a = '1' , b = '2' , c = '3'), Row(a= '4' , b = '5' , c = '6')]]

我想得到['3' , '6']作为我的结果，它是列表中每个元素的第三个元素。

Answer 1

不确定完全理解你在问什么。 让我们试试这个答案：

from pyspark.sql import Row

a = [[Row(a = '1' , b = '2' , c = '3'), Row(a= '4' , b = '5' , c = '6')]]

result = [j[2] for i in a for j in i]

print(result)
>['3', '6']