Spark MLlib之使用Breeze操作矩阵向量

在使用Breeze 库时,需要导入相关包:

import breeze.linalg._
import breeze.numerics._

Breeze创建函数

//全0矩阵
DenseMatrix.zeros[Double](,) res0: breeze.linalg.DenseMatrix[Double] =
0.0 0.0
0.0 0.0
0.0 0.0 //全0向量
DenseVector.zeros[Double]() res1: breeze.linalg.DenseVector[Double] = DenseVector(0.0, 0.0) //全1向量
DenseVector.ones[Double]() res2: breeze.linalg.DenseVector[Double] = DenseVector(1.0, 1.0) //按数值填充向量
DenseVector.fill[Double](, ) res3: breeze.linalg.DenseVector[Double] = DenseVector(2.0, 2.0, 2.0) //生成随机向量
DenseVector.range(, , )
DenseVector.rangeD(, , )
DenseVector.rangeF(, , ) res4: breeze.linalg.DenseVector[Int] = DenseVector(, , , )
res5: breeze.linalg.DenseVector[Double] = DenseVector(1.0, 3.0, 5.0, 7.0)
res6: breeze.linalg.DenseVector[Float] = DenseVector(1.0, 3.0, 5.0, 7.0) //单位矩阵
DenseMatrix.eye[Double]() res7: breeze.linalg.DenseMatrix[Double] =
1.0 0.0 0.0 0.0
0.0 1.0 0.0 0.0
0.0 0.0 1.0 0.0
0.0 0.0 0.0 1.0 //对角矩阵
diag(DenseVector(3.0, 4.0, 5.0)) res8: breeze.linalg.DenseMatrix[Double] =
3.0 0.0 0.0
0.0 4.0 0.0
0.0 0.0 5.0 //按照行创建矩阵
DenseMatrix((4.0, 5.0, 6.0), (7.0, 8.0, 9.0)) res9: breeze.linalg.DenseMatrix[Double] =
4.0 5.0 6.0
7.0 8.0 9.0 //按照行创建向量
DenseVector((4.0, 5.0, 6.0, 7.0, 8.0, 9.0)) res10: breeze.linalg.DenseVector[(Double, Double, Double, Double, Double, Double)] = DenseVector((4.0,5.0,6.0,7.0,8.0,9.0)) //向量转置
DenseVector((4.0, 5.0, 6.0, 7.0, 8.0, 9.0)).t res11: breeze.linalg.Transpose[breeze.linalg.DenseVector[(Double, Double, Double, Double, Double, Double)]] = Transpose(DenseVector((4.0,5.0,6.0,7.0,8.0,9.0))) //从函数创建向量
DenseVector.tabulate()(i => i*i)
DenseVector.tabulate( to )(i => i*i) res12: breeze.linalg.DenseVector[Int] = DenseVector(, , , , )
res13: breeze.linalg.DenseVector[Int] = DenseVector(, , , , , ) //从函数创建矩阵
DenseMatrix.tabulate(, ){ case (i, j) => i*i+j*j } res14: breeze.linalg.DenseMatrix[Int] = //从数组创建向量
new DenseVector[Double](Array(2.0, 5.0, 8.0)) res15: breeze.linalg.DenseVector[Double] = DenseVector(2.0, 5.0, 8.0) //从数组创建矩阵
new DenseMatrix[Double](, , Array(1.0, 4.0, 7.0, 3.0, 6.0, 9.0)) res16: breeze.linalg.DenseMatrix[Double] =
1.0 3.0
4.0 6.0
7.0 9.0 //0 到 1的随机向量
DenseVector.rand(, Rand.uniform)
DenseVector.rand(, Rand.gaussian) res17: breeze.linalg.DenseVector[Double] = DenseVector(0.30960687979350654, 0.5779984012083466, 0.4880956198283952, 0.1013947992922748, 0.19635570812305936, 0.8533170989347008, 0.6619843996111201, 0.03131533370356321, 0.5430592884856604)
res18: breeze.linalg.DenseVector[Double] = DenseVector(0.48361471134641176, -1.734778260551877, -0.7319505628964431, 0.19971267958211184, -1.033191008131693, -1.7961545888066046, 0.2364555601503527, 0.22843047924270285, 1.7288956723034343) //0 到 1的随机矩阵
DenseMatrix.rand(, , Rand.uniform)
DenseMatrix.rand(, , Rand.gaussian) res19: breeze.linalg.DenseMatrix[Double] = 0.11270960774886585 0.19871332589909851
0.5581898434134047 0.8295064603050235
0.8692650535288642 0.4015512971620494
res20: breeze.linalg.DenseMatrix[Double] = 0.712041684728872 2.7007736007506216
0.053520407807479485 0.19044772577405517
-0.7370909025873376 -1.024737052742153

Breeze元素访问

val a = new DenseVector[Int](Array( to : _*))

a: breeze.linalg.DenseVector[Int] = DenseVector(, , , , , , , , , , )

//指定位置
a() res21: Int = //向量子集
a( to ) res22: breeze.linalg.DenseVector[Int] = DenseVector(, , , ) //按照指定步长取子集
a( to by -) res23: breeze.linalg.DenseVector[Int] = DenseVector(, , , , , ) //指定开始位置至结尾
a( to -) res24: breeze.linalg.DenseVector[Int] = DenseVector(, , , , , , , , , ) //最后一个元素
a(-) res25: Int = val m = DenseMatrix((1.0, 2.0, 3.0), (4.0, 5.0, 6.0)) m: breeze.linalg.DenseMatrix[Double] =
1.0 2.0 3.0
4.0 5.0 6.0 //指定位置
m(, ) res26: Double = 2.0 //矩阵指定列
m(::, )

val a = new DenseVector[Int](Array(10 to 20: _*))

a: breeze.linalg.DenseVector[Int] = DenseVector(10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20)

//指定位置
a(0)

res21: Int = 10

//向量子集
a(1 to 4)

res22: breeze.linalg.DenseVector[Int] = DenseVector(11, 12, 13, 14)

//按照指定步长取子集
a(5 to 0 by -1)

res23: breeze.linalg.DenseVector[Int] = DenseVector(15, 14, 13, 12, 11, 10)

//指定开始位置至结尾
a(1 to -1)

res24: breeze.linalg.DenseVector[Int] = DenseVector(11, 12, 13, 14, 15, 16, 17, 18, 19, 20)

//最后一个元素
a(-1)

res25: Int = 20

val m = DenseMatrix((1.0, 2.0, 3.0), (4.0, 5.0, 6.0))

m: breeze.linalg.DenseMatrix[Double] =
1.0 2.0 3.0
4.0 5.0 6.0

//指定位置
m(0, 1)

res26: Double = 2.0

//矩阵指定列
m(::, 1)
res27: breeze.linalg.DenseVector[Double] = DenseVector(2.0, 5.0)

Breeze元素操作

//调整矩阵形状
m.reshape(, ) res28: breeze.linalg.DenseMatrix[Double] =
1.0 6.0 11.0
5.0 10.0 4.0
9.0 3.0 8.0
2.0 7.0 12.0 //矩阵转成向量
m.toDenseVector res29: breeze.linalg.DenseVector[Double] = DenseVector(1.0, 5.0, 9.0, 2.0, 6.0, 10.0, 3.0, 7.0, 11.0, 4.0, 8.0, 12.0) //复制下三角
lowerTriangular(m) res30: breeze.linalg.DenseMatrix[Double] =
1.0 0.0 0.0
5.0 6.0 0.0
9.0 10.0 11.0 //复制上三角
upperTriangular(m) res31: breeze.linalg.DenseMatrix[Double] =
1.0 2.0 3.0
0.0 6.0 7.0
0.0 0.0 11.0 //矩阵复制
m.copy res32: breeze.linalg.DenseMatrix[Double] =
1.0 2.0 3.0 4.0
5.0 6.0 7.0 8.0
9.0 10.0 11.0 12.0 //取对角线元素
diag(upperTriangular(m)) res33: breeze.linalg.DenseVector[Double] = DenseVector(1.0, 6.0, 11.0) //子集赋数值
a( to ) :=
a res34: breeze.linalg.DenseVector[Int] = DenseVector(, , , )
res35: breeze.linalg.DenseVector[Int] = DenseVector(, , , , , , , , , , ) //子集赋向量
a( to ) := DenseVector(, , , )
a res36: breeze.linalg.DenseVector[Int] = DenseVector(, , , )
res37: breeze.linalg.DenseVector[Int] = DenseVector(, , , , , , , , , , ) //矩阵赋值
m( to , to ) := 0.0
m res38: breeze.linalg.DenseMatrix[Double] =
0.0 0.0
0.0 0.0
res39: breeze.linalg.DenseMatrix[Double] =
1.0 2.0 3.0 4.0
5.0 0.0 0.0 8.0
9.0 0.0 0.0 12.0 //矩阵列赋值
m(::, ) := 5.0
m res40: breeze.linalg.DenseVector[Double] = DenseVector(5.0, 5.0, 5.0)
res41: breeze.linalg.DenseMatrix[Double] =
1.0 2.0 5.0 4.0
5.0 0.0 5.0 8.0
9.0 0.0 5.0 12.0 val a1 = DenseMatrix((1.0, 2.0, 3.0), (4.0, 5.0, 6.0))
val a2 = DenseMatrix((7.0, 8.0, 9.0), (10.0, 11.0, 12.0)) a1: breeze.linalg.DenseMatrix[Double] =
1.0 2.0 3.0
4.0 5.0 6.0
a2: breeze.linalg.DenseMatrix[Double] =
7.0 8.0 9.0
10.0 11.0 12.0 //垂直连接矩阵
DenseMatrix.vertcat(a1, a2) res42: breeze.linalg.DenseMatrix[Double] =
1.0 2.0 3.0
4.0 5.0 6.0
7.0 8.0 9.0
10.0 11.0 12.0 //横向连接矩阵
DenseMatrix.horzcat(a1, a2) res43: breeze.linalg.DenseMatrix[Double] =
1.0 2.0 3.0 7.0 8.0 9.0
4.0 5.0 6.0 10.0 11.0 12.0 //向量连接
DenseVector.vertcat(DenseVector(, , ), DenseVector(, , ))
DenseVector.horzcat(DenseVector(, , ), DenseVector(, , )) res44: breeze.linalg.DenseVector[Int] = DenseVector(, , , , , )
res45: breeze.linalg.DenseMatrix[Int] =

Breeze数值计算函数

//元素加法
a1 + a2 res46: breeze.linalg.DenseMatrix[Double] =
8.0 10.0 12.0
14.0 16.0 18.0 //元素乘法
a1 :* a2 res47: breeze.linalg.DenseMatrix[Double] =
7.0 16.0 27.0
40.0 55.0 72.0 //元素除法
a1 :/ a2 res48: breeze.linalg.DenseMatrix[Double] =
0.14285714285714285 0.25 0.3333333333333333
0.4 0.45454545454545453 0.5 //元素比较
a1 :< a2 res49: breeze.linalg.DenseMatrix[Boolean] =
true true true
true true true //元素相等
a1 :== a2 res50: breeze.linalg.DenseMatrix[Boolean] =
false false false
false false false //元素追加
a1 :+=2.0 res51: breeze.linalg.DenseMatrix[Double] =
3.0 4.0 5.0
6.0 7.0 8.0 //元素追乘
a1 :*=2.0 res52: breeze.linalg.DenseMatrix[Double] =
6.0 8.0 10.0
12.0 14.0 16.0 //向量点积
DenseVector(, , , ) dot DenseVector(, , , ) res53: Int = //元素最大值
max(a1) res54: Double = 16.0 //元素最小值
min(a1) res55: Double = 6.0 //元素最大值的位置
argmax(a1) res56: (Int, Int) = (,) //元素最小值的位置
argmin(a1) res57: (Int, Int) = (,)

Breeze求和函数

val m1 = DenseMatrix((1.0, 2.0, 3.0, 4.0), (5.0, 6.0, 7.0, 8.0), (9.0, 10.0, 11.0, 12.0))

m1: breeze.linalg.DenseMatrix[Double] =
1.0 2.0 3.0 4.0
5.0 6.0 7.0 8.0
9.0 10.0 11.0 12.0 //元素求和
sum(m1) res58: Double = 78.0 //每一列求和
sum(m1, Axis._0) res59: breeze.linalg.DenseMatrix[Double] = 15.0 18.0 21.0 24.0 //每一行求和
sum(m1, Axis._1) res60: breeze.linalg.DenseVector[Double] = DenseVector(10.0, 26.0, 42.0) //对角线元素和
trace(lowerTriangular(m1)) res61: Double = 18.0 //累积和
val a3 = new DenseVector[Int](Array( to : _*))
accumulate(a3) a3: breeze.linalg.DenseVector[Int] = DenseVector(, , , , , , , , , , )
res62: breeze.linalg.DenseVector[Int] = DenseVector(, , , , , , , , , , )

Breeze布尔函数

val c = DenseVector(true, false, true)
val d = DenseVector(false, true, true)
//元素与操作
c :& d res63: breeze.linalg.DenseVector[Boolean] = DenseVector(false, false, true) //元素或操作
c :| d res64: breeze.linalg.DenseVector[Boolean] = DenseVector(true, true, true) //元素非操作
!c res65: breeze.linalg.DenseVector[Boolean] = DenseVector(false, true, false) val e = DenseVector[Int](-, , ) e: breeze.linalg.DenseVector[Int] = DenseVector(-, , ) //存在非零元素
any(e) res66: Boolean = true //所有元素非零
all(e) res67: Boolean = false

Breeze线性代数函数

val f = DenseMatrix((1.0, 2.0, 3.0), (4.0, 5.0, 6.0), (7.0, 8.0, 9.0))
val g = DenseMatrix((1.0, 1.0, 1.0), (1.0, 1.0, 1.0), (1.0, 1.0, 1.0)) f: breeze.linalg.DenseMatrix[Double] =
1.0 2.0 3.0
4.0 5.0 6.0
7.0 8.0 9.0
g: breeze.linalg.DenseMatrix[Double] =
1.0 1.0 1.0
1.0 1.0 1.0
1.0 1.0 1.0 //线性求解,AX = B,求解X
f \ g res68: breeze.linalg.DenseMatrix[Double] =
-2.5 -2.5 -2.5
4.0 4.0 4.0
-1.5 -1.5 -1.5 //转置
f.t res69: breeze.linalg.DenseMatrix[Double] =
1.0 4.0 7.0
2.0 5.0 8.0
3.0 6.0 9.0 //求特征值
det(f) res70: Double = 6.661338147750939E-16 //求逆
inv(f) res71: breeze.linalg.DenseMatrix[Double] =
-4.503599627370499E15 9.007199254740992E15 -4.503599627370495E15
9.007199254740998E15 -1.8014398509481984E16 9.007199254740991E15
-4.503599627370498E15 9.007199254740992E15 -4.5035996273704955E15 //求伪逆
pinv(f) res72: breeze.linalg.DenseMatrix[Double] =
-3.7720834019330525E14 7.544166803866101E14 -3.77208340193305E14
7.544166803866094E14 -1.5088333607732208E15 7.544166803866108E14
-3.772083401933041E14 7.544166803866104E14 -3.772083401933055E14 //特征值和特征向量
eig(f) res73: breeze.linalg.eig.DenseEig =
Eig(DenseVector(16.116843969807043, -1.1168439698070427, -1.3036777264747022E-15),
DenseVector(0.0, 0.0, 0.0),
-0.23197068724628617 -0.7858302387420671 0.40824829046386363
-0.5253220933012336 -0.08675133925662833 -0.816496580927726
-0.8186734993561815 0.61232756022881 0.4082482904638625 ) //奇异值分解
val svd.SVD(u,s,v) = svd(g) u: breeze.linalg.DenseMatrix[Double] =
-0.5773502691896255 -0.5773502691896257 -0.5773502691896256
-0.5773502691896256 -0.2113248654051871 0.7886751345948126
-0.5773502691896256 0.7886751345948129 -0.21132486540518708
s: breeze.linalg.DenseVector[Double] = DenseVector(3.0000000000000004, 0.0, 0.0)
v: breeze.linalg.DenseMatrix[Double] =
-0.5773502691896256 -0.5773502691896257 -0.5773502691896256
0.0 -0.7071067811865474 0.7071067811865477
0.816496580927726 -0.4082482904638629 -0.4082482904638628 //求矩阵的秩
rank(f) res74: Int = //矩阵长度
f.size res75: Int = //矩阵行数
f.rows res76: Int = //矩阵列数
f.cols res77: Int =

Breeze取整函数

val h = DenseVector(-1.2, 0.7, 2.3)

h: breeze.linalg.DenseVector[Double] = DenseVector(-1.2, 0.7, 2.3)

//四舍五入
round(h) res78: breeze.linalg.DenseVector[Long] = DenseVector(-, , ) //大于它的最小整数
ceil(h) res79: breeze.linalg.DenseVector[Double] = DenseVector(-1.0, 1.0, 3.0) //小于它的最大整数
floor(h) res80: breeze.linalg.DenseVector[Double] = DenseVector(-2.0, 0.0, 2.0) //符号函数
signum(h) res81: breeze.linalg.DenseVector[Double] = DenseVector(-1.0, 1.0, 1.0) //取正数
abs(h) res82: breeze.linalg.DenseVector[Double] = DenseVector(1.2, 0.7, 2.3)
上一篇:【dp】饥饿的牛


下一篇:POJ 2411 Mondriaan's Dream (状压DP,骨牌覆盖,经典)