如何從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);