Как извлечь значения из вектора с помощью udf в pyspark

Question

Я не могу получить доступ к элементам внутри вектора, используя UDF в pyspark

Я уже пытался использовать UDF, однако я не могу извлечь нужные элементы

код:

from pyspark.sql.functions import udf
from pyspark.sql.types import FloatType

firstelement=udf(lambda v:float(v[0]),FloatType())
trainingData.select(firstelement('features')).show()

Как получить доступ к 1496 из (8255, [1496,3460 ....