Tenho um spark dataframe com linhas como -

1   |   [a, b, c]
2   |   [d, e, f]
3   |   [g, h, i]

gora, quero manter apenas os 2 primeiros elementos da coluna da matri

1   |   [a, b]
2   |   [d, e]
3   |   [g, h]

Como isso pode ser alcançado?

ota - Lembre-se de que não estou extraindo um único elemento da matriz aqui, mas uma parte da matriz que pode conter vários elemento