如何从Java中的SparseVectors的RDD创建数据框

Question

我试图将RDD的SparseVector转换为DataFrame。我在Scala和Python这样做过，但从未在Java，的确可以找到答案here。

我试图找到涵盖此主题的示例，但找不到任何示例。尽管如此，它看起来和Scala一样，但我无法复制它。

Answer 1

我终于可以实现它了。将用于转换SparseVector的适当架构必须设置为dataType a new VectorUDT，请注意，如果按照示例，它们通常使用数据类型。 东西，所以它很艰难。

List<StructField> fields = new ArrayList<>(); 
StructField field = DataTypes.createStructField("features", new VectorUDT(), true); 

fields.add(field); 

StructType schema = DataTypes.createStructType(fields);