Java Matrix Sum sum(double[][] X, int axis)

Description

sum

License

Open Source License

Declaration

public static double[] sum(double[][] X, int axis) 

Method Source Code

//package com.java2s;
/*******************************************************************************
 * Copyright (c) 2014, 2015 IBM Corporation
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE./*  w w w . j  av  a 2s . c o m*/
 *******************************************************************************/

public class Main {
    public static double sum(double[] x) {
        int n = x.length;
        double s = 0;
        for (int i = 0; i < n; i++) {
            s += x[i];
        }
        return s;
    }

    public static double[] sum(double[][] X, int axis) {
        if (axis == 0) {
            return sumAxis0(X);
        } else if (axis == 1) {
            return sumAxis1(X);
        } else {
            String message = "axis>=2 is not supported in sum method.";
            throw new RuntimeException(message);
        }
    }

    static double[] sumAxis0(double[][] X) {
        int m = X[0].length;
        double[] array = new double[m];
        for (int j = 0; j < m; j++) {
            double[] x = column(X, j);
            array[j] = sum(x);
        }
        return array;
    }

    static double[] sumAxis1(double[][] X) {
        int n = X.length;
        double[] array = new double[n];
        for (int i = 0; i < n; i++) {
            double[] x = row(X, i);
            array[i] = sum(x);
        }
        return array;
    }

    public static double[] column(double[][] array, int j) {
        int rows = array.length;
        double[] colvec = new double[rows];
        for (int i = 0; i < rows; i++) {
            colvec[i] = array[i][j];
        }
        return colvec;
    }

    public static double[] row(double[][] array, int i) {
        return array[i];
    }
}

Related

1. sum(double[][] error)
2. sum(double[][] input, int column)
3. sum(double[][] input, int column)
4. sum(double[][] kernel1, double[][] kernel2)
5. sum(double[][] o)
6. sum(float[][] a1, float[][] a2)
7. sum(int M[][])
8. sum(int[][][] X, int[] coords)
9. sumArrayDim(double[][] array, int dimToSummarize)