df = sc.parallelize([('assert', Vectors.dense([1, 2, 3]))]).toDF(['word',
    'vector'])

def extract(row):
    return (row.word,) + tuple(float(x) for x in row.vector.values)
df.rdd.map(extract).toDF(['word'])
Как разбить плотный вектор на столбцы с помощью pyspark