Java使用math3实现矩阵行求和(求行均值)

爱被打了一巴掌 2022-04-15 05:06 318阅读 0赞

在写算法时，有时会用到行求和或列求和，下面将以java中的math简单实现这一操作：

例如，我们遇到这样多个一维数组：

double a[] = new double[]{ 1,2,3,4};
    double b[] = new double[]{ 1,2,3,3};
    ...

实现每个数组对应列上的数字相加，然后求均值，即要得到的结果为：

1.0,2.0,3.0,3.5

下面为math3操作的方式：

package deal;
    
    import java.util.ArrayList;
    import java.util.List;
    import org.apache.commons.math3.linear.Array2DRowRealMatrix;
    import org.apache.commons.math3.linear.RealMatrix;
    
    public class TestDeal { 
    
    	public static void main(String[] args) { 
    		double a[] = new double[]{ 1,2,3,4};
    		double aa[][] = TwoArry(a);
    		double b[] = new double[]{ 1,2,3,3};
    		double bb[][] = TwoArry(b);
    		RealMatrix matrix_1 = new Array2DRowRealMatrix(aa);
    		RealMatrix matrix_2 = new Array2DRowRealMatrix(bb);
    		System.out.println(matrix_1);
    		List<RealMatrix> wMatrixList = new ArrayList<RealMatrix>();
    		wMatrixList.add(matrix_1);
    		wMatrixList.add(matrix_2);
    		RealMatrix realMatrix = sum(wMatrixList);
    		System.out.println(realMatrix);
    
    	}
    	private static RealMatrix sum(List<RealMatrix> wMatrixList){ 
    		//定义一个新矩阵
    		RealMatrix suMatrix= new Array2DRowRealMatrix( wMatrixList.get(0).getRowDimension(), wMatrixList.get(0).getColumnDimension());
    		//多个矩阵相加的操作
    		for( RealMatrix re: wMatrixList){ 
    			suMatrix = suMatrix.add(re);
    		}
    		//返回均值
    		return suMatrix.scalarMultiply(1.0/wMatrixList.size());
    	}
    	//一维数组转化为二维数组
    	public static double[][] TwoArry(double[] onedouble){ 
    		double[][] arr = new double[1][onedouble.length];
    		for (int i = 0; i < onedouble.length; i++) { 
    			arr[0][i] = onedouble[i];
    		}
    		return arr;
    	}
    }

程序输出结果为：  
![在这里插入图片描述][20181121212712355.png]

[20181121212712355.png]: /images/20220415/9cf1f3aaec9348b9b8f0529f7780c5bd.png