SparkMllib基本数据类型

package cn.li.localvector

import org.apache.spark.mllib.linalg.{Vector, Vectors}

/**
  * 2.SparkMllib基本数据类型
  * 2021-01-23
  */
object testVector {
  def main(args: Array[String]): Unit = {
    val vd:Vector = Vectors.dense(2,0,6)//建立密集向量
    println(vd)
    println(vd(0))
    println(vd(1))
    println(vd(2))//打印稀疏向量第3个值
//    val vs: Vector = Vectors.sparse(4,Array(0,1,2,3),Array(9,5,2,7))//建立稀疏向量
    //第一个参数4代表输入数据的大小，一般要求大于等于输入的数据值，第二个参数是数据下标，第三个参数是数据值
//    println(vs)//打印稀疏向量第3个值
//    println(vs(2))//打印稀疏向量第3个值
    println("-----------------------")
    val vs:Vector = Vectors.sparse(4,Array(0,3),Array(9,7))
    println(vs)
    println(vs(0))
    println(vs(3))
    /**
      * 9.0
        7.0
     */
    println("-----------------------")
    //通过指定其非零条目来创建稀疏向量(1.0,0.0,3.0)
    val sv2:Vector = Vectors.sparse(3,Seq((0,1.0),(2,3.0)))
    println(sv2(0))
    println(sv2(1))
    println(sv2(2))


  }
}

[2.0,0.0,6.0]
2.0
0.0
6.0
-----------------------
(4,[0,3],[9.0,7.0])
9.0
7.0
-----------------------
1.0
0.0
3.0