我会把问题分解成几个步骤。

# First we set it up

data = [(1,2,3,4,5,6,7,8,9,10)] # one row
rdd = spark.sparkContext.parallelize(data)
rdd.collect()

# [(1, 2, 3, 4, 5, 6, 7, 8, 9, 10)]

接下来我们需要一个从元组中弹出一个值并使其成为键的函数。

def key_elem_to_rest(key_index, tup):
  l = list(tup)
  key = l.pop(key_index)
  return {key: tuple(l)}

下一步我们在 map ```
rdd.map(lambda x: key_elem_to_rest(0, x)).collect() # index = 0

[{1: (2, 3, 4, 5, 6, 7, 8, 9, 10)}]

rdd.map(lambda x: key_elem_to_rest(5, x)).collect() # index = 5

[{6: (1, 2, 3, 4, 5, 7, 8, 9, 10)}]

您可以尝试使用以下方法：

rdd.filter(lambda x: x[0] != x[9]).map(lambda x: (x[9], [x[:-1]]))

这是在检查x[9]是否不是一个键，它将它作为一个键，而rest作为值。

我终于自己弄明白了。

units_rdd1 = units_rdd.map(lambda x: (x[9], list(x[0:9]+x[10:])))