Spark机器学习库-矩阵or向量的获取

1、数据准备

    val vector = DenseVector(1,2,4,4,5,6)
    val matrix = new DenseMatrix[Int](2,3,Array(1,2,3,4,5,6))

2、获取指定位置的元素

//访问指定位置的元素
    println(vector.valueAt(0))
    println(matrix(0,0))

3、获取矩阵或者数组的子集

    println(vector(0 to 4))
    println(matrix(0 to 1,0))

4、按照指定步长取子集

println(vector(vector.length-1 to 0 by -1))

5、从指定位置开始一直到结尾

println(vector(0 to -1))

6、最后一个元素

println(vector(-1))

矩阵指定的列

   println(matrix(::,-1))

7、矩阵指定的行

println(matrix(-1,::))

8、Code

/**
  * Created by HeartisTiger on 2019/7/7
  */
object Demo02 {
  def main(args: Array[String]): Unit = {

    val vector = DenseVector(1, 2, 4, 4, 5, 6)
    val matrix = new DenseMatrix[Int](2, 3, Array(1, 2, 3, 4, 5, 6))

    //访问指定位置的元素
    println(vector.valueAt(0))
    println(matrix(0, 0))
    //访问矩阵的子集
    println(vector(0 to 4))
    println(matrix(0 to 1, 0))
    //按照指定步长取子集
    println(vector(vector.length - 1 to 0 by -1))

    //指定开始位置一直到结尾
    println(vector(0 to -1))

    //最后一个元素
    println(vector(-1))

    //矩阵指定列
    println(matrix)
    println(matrix(::, -1))

    //矩阵指定行
    println(matrix(-1, ::))

  }
}