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

【问题标题】：get n-th element from a tuple inside array pySpark从数组pySpark中的元组中获取第n个元素
【发布时间】：2021-12-22 02:04:50
【问题描述】：

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

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

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

【问题讨论】：

您想获得 C 列吗？ df.select("C")

标签： python arrays list pyspark tuples

【解决方案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']

【讨论】：