java - 实数矩阵与复数矩阵相乘

Question

我对 Java 还很陌生，我面临一个问题，我相信它可以很容易掌握。

我正在生成一个链接到 Apache - Commons Math 库的项目。

在项目中，我使用了大量的RealMatrix对象。我有一个方法，工作原理如下

public static RealMatrix DistCalc(RealMatrix YCoord, RealMatrix ZCoord){
        RealMatrix Distance = new Array2DRowRealMatrix(YCoord.getRowDimension(),ZCoord.getRowDimension());
        for(int ii = 0; ii < YCoord.getRowDimension(); ii++){
            for(int jj = 0; jj < ZCoord.getRowDimension(); jj++){
                Distance.setEntry(ii,jj,Math.sqrt((YCoord.getEntry(ii, 0) - YCoord.getEntry(jj, 0))*(YCoord.getEntry(ii, 0) - YCoord.getEntry(jj, 0)) + (ZCoord.getEntry(jj, 0) - ZCoord.getEntry(ii, 0))*(ZCoord.getEntry(jj, 0) - ZCoord.getEntry(ii, 0))));
            }
        }        
        return Distance;
    }

另一个生成特定的复杂矩阵，

// Define the random phase for the u- component
    public static Complex[][] RandPhi(int N, int nFFT){
        Complex[][] nn_u = new Complex[N][nFFT];        
        for(int ii = 0; ii < N; ii++){
            for(int jj = 0; jj < nFFT; jj++){
                nn_u[ii][jj] = new Complex(Math.cos(new Random().nextDouble()*2*Math.PI),Math.sin(new Random().nextDouble()*2*Math.PI));
            }
        }
        return nn_u;
    }

现在，我想按列将 RealMatrix Distance 与 Complex 矩阵 nn_u 相乘:最后我应该想出一个Complex[N][nFFT]矩阵。

您介意透露一些信息吗？

Answer 1

我建议您基于 RealMatrix 接口(interface)创建自己的 ComplexMatrix 接口(interface)，然后基于Array2DRowRealMatrix 类。要创建该类，只需 download the source code ，更改类名，将 double data[][] 更改为 Complex data[][]，然后更新对 data 的所有引用。

要么创建一个接受 RealMatrix 的 ComplexMatrix 构造函数，要么包含一个带有 RealMatrix 的 multiply 方法参数。

Commons 应该拥有您需要的所有方法，您可能只需要稍微调整它们的参数/返回类型。