Spark Scalaを使用して1つの列データをベクトルに変換する方法

Question

Spark、Scalaを使用してデータを処理しています。私は1つの質問が把握できなかった。

data 
1 
2 
3 
4 
5 

私は、単一のベクター にそれをしたい[1.0,2.0,3.0,4.0,5.0]

私はそれを実装するにはどうすればよい

：私は1つの列であるデータフレームを、持っていますか？私は df.collect().toVectorまたはrdd.foreachを試しましたが、たびにベクトル[1,0]、[2.0]、[3.0]、[4.0]、[5.0]の配列を返します。

Answer 1

これは、collectデータフレームを使用すると、行の配列を取得するためです。行オブジェクトから値を抽出する必要があります。

df.collect().map(x => x.getDouble(0)).toVector