3 Spark机器学习 spark MLlib 矩阵向量、矩阵运算Breeze库-2

package breeze

import breeze.linalg.{DenseMatrix, argmax, max}

object Math {
    def main(args: Array[String]): Unit = {
        val a = DenseMatrix((1, 2, 3), (4, 5, 6))
        val b = DenseMatrix((1, 1, 1), (2, 2, 2))
        println(a + b)
        println("-------------")
        //每个元素相乘
        println(a *:* b)
        println("-------------")
        //每个元素相除
        println(a /:/ b)
        println("-------------")
        //比较每个元素的大小
        println(a <:< b)
        println("-------------")
        //是否相等
        println(a :== b)
        println("-------------")
        //追加，每个元素都加1
        println(a :+= 1)
        println("-------------")
        //追乘，每个元素都乘2
        println(a :*= 2)
        println("-------------")
        val x = DenseMatrix((1, 2, 3), (4, 5, 6))
        //元素最大值
        println(max(x))
        println("-------------")
        //最大值及其位置
        println(argmax(x))
    }

}

import breeze.linalg.{DenseMatrix, accumulate, sum, trace}

object Sum {
    def main(args: Array[String]): Unit = {
        val a = DenseMatrix((1, 2, 3), (4, 5, 6))
        println(sum(a))
        println("-----------")
        //对第2列求和
        println(sum(a(::, 1)))
        println("-----------")
        //对第1行求和
        println(sum(a(0, ::)))
        println("-----------")
        //对角线求和，必须是正方形矩阵
        val b = DenseMatrix((1, 2, 3), (4, 5, 6), (7, 8, 9))
        println(trace(b))
    }

}