使用Colt库的简单矩阵运算(2D和3D)

时间:2013-10-17 10:31:41

标签: java matrix add scalar colt

我想在我的代码中执行简单的矩阵运算,并使用Colt库

(见这里:http://acs.lbl.gov/software/colt/api/index.html

我希望例如添加/减去/乘以矩阵,对标量的每个单元格添加/减去/乘/除标量......但是这个库中似乎没有这样的函数。

但是,我发现了这条评论:https://stackoverflow.com/a/10815643/2866701

如何使用 assign()命令在我的代码中执行这些简单的操作?

2 个答案:

答案 0 :(得分:3)

Colt在assign()方法下提供了更通用的框架。例如,如果要为矩阵的每个单元格添加标量,可以执行以下操作:

double scalar_to_add  = 0.5;
DoubleMatrix2D matrix = new DenseDoubleMatrix2D(10, 10); // creates an empty 10x10 matrix
matrix.assign(DoubleFunctions.plus(scalar_to_add)); // adds the scalar to each cell

标准函数在DoubleFunctions类中作为静态方法提供。其他人需要由你写。

如果您想要添加矢量而不是仅添加标量值,则assign()的第二个参数必须是DoubleDoubleFunction。例如,

DoubleDoubleFunction plus = new DoubleDoubleFunction() {
    public double apply(double a, double b) { return a+b; }
};    
DoubleMatrix1D aVector = ... // some vector
DoubleMatrix1D anotherVector = ... // another vector of same size
aVector.assign(anotherVector, plus); // now you have the vector sum

答案 1 :(得分:0)

为什么不试试la4j(线性代数for Java)?它易于使用:

Matrix a = new Basci2DMatrix(new double[][]{
    { 1.0, 2.0 },
    { 3.0, 4.0 }
});

Matrix b = new Basci2DMatrix(new double[][]{
    { 5.0, 6.0 },
    { 7.0, 8.0 }
});

Matrix c = a.multiply(b); // a * b
Matrix d = a.add(b); // a + b
Matrix e = a.subtract(b); // a - b

还有transform()方法类似于Colt的assign()。它可以用作以下内容:

Matrix f = a.transform(Matrices.INC_MATRIX); // inreases each cell by 1
Matrix g = a.transform(Matrices.asDivFunction(2)); // divides each cell by 2
// you can define your own function
Matrix h = a.transform(new MatrixFunction {
  public double evaluate(int i, int j, int value) {
    return value * Math.sqrt(i + j);
  }
});

但它仅适用于 2D 矩阵。