如何在 Scala 中将 Spark Dense Vector 转换为 String 并返回 Vector？答案

【问题标题】：How to convert Spark Dense Vector to String and back to Vector in Scala?如何在 Scala 中将 Spark Dense Vector 转换为 String 并返回 Vector？
【发布时间】：2017-11-08 17:58:19
【问题描述】：

我有密集向量，我想将向量转换为字符串（以保存 CSV）并在加载时将字符串转换回密集向量。

更多细节

val dense_vec = Vectors.dense(1.0, 2.0, 3.0)
dense_vec: org.apache.spark.mllib.linalg.Vector = [1.0,2.0,3.0]

val str_dense_vec = dense_vec.toString
str_dense_vec: String = [1.0,2.0,3.0]

我想将str_dense_vec 类型为String 转换为org.apache.spark.mllib.linalg.Vector

【问题讨论】：

标签： scala csv apache-spark export-to-csv

【解决方案1】：

您可以从字符串创建双精度数组，然后使用org.apache.spark.mllib.linalg.Vector 的dense 方法。

Vectors.dense(str_dense_vec.drop(1).dropRight(1).split(',').map(_.toDouble))

linkapi.

【讨论】：