Matrix Operations in Java

• Download source code - 14.8 KB

Introduction

This article introduces some basic methods in Java for matrix additions, multiplications, inverse, transpose, and other relevant operations. The matrix operations are explained briefly and external links are given for more details. The main functions are given as static utility methods. All methods in this article are unit tested and the test codes are part of the attached files.

As suggested by a member (i.e., César de Souza), the matrix decomposition methods such as Cholesky Decomposition and LU decomposition are more common in matrix operations.

Using the Code

Matrix is a two dimensional array of numbers. I define Matrix in Java using three parameters; i.e., number of rows (nrows), number of columns (ncols), and the data as an array of doubles.

public class Matrix {

    private int nrows;
    private int ncols;
    private double[][] data;

    public Matrix(double[][] dat) {
        this.data = dat;
        this.nrows = dat.length;
        this.ncols = dat[0].length;
    }

    public Matrix(int nrow, int ncol) {
        this.nrows = nrow;
        this.ncols = ncol;
        data = new double[nrow][ncol];
    }

Matrices are fundamental in mathematics and their operations are vital in quantitative subjects. In separate articles, I will use these functions for statistical modeling.

Transpose

Transpose of a matrix is produced by swapping the rows with columns.

public static Matrix transpose(Matrix matrix) {
    Matrix transposedMatrix = new Matrix(matrix.getNcols(), matrix.getNrows());
    for (int i=0;i<matrix.getNrows();i++) {
        for (int j=0;j<matrix.getNcols();j++) {
            transposedMatrix.setValueAt(j, i, matrix.getValueAt(i, j));
        }
    }
    return transposedMatrix;
} 

For more information about transpose of a matrix, visit this link.

Determinant of a Square Matrix

A square matrix has an equal number of rows and columns. For these matrices, the following method can be used to calculate the determinant. We will use this function later in this article to find the inverse of a matrix.

public static double determinant(Matrix matrix) throws NoSquareException {
    if (!matrix.isSquare())
        throw new NoSquareException("matrix need to be square.");
    if (matrix.size() == 1) {
	return matrix.getValueAt(0, 0);
    }
    if (matrix.size()==2) {
        return (matrix.getValueAt(0, 0) * matrix.getValueAt(1, 1)) - 
                         ( matrix.getValueAt(0, 1) * matrix.getValueAt(1, 0));
    }
    double sum = 0.0;
    for (int i=0; i<matrix.getNcols(); i++) {
        sum += changeSign(i) * matrix.getValueAt(0, i) * determinant(createSubMatrix(matrix, 0, i));
    }
    return sum;
} 

changeSign(i) is a method that returns 1 if i is even and -1 otherwise. More information about determinants are given here.

The above method is a recursive function that breaks the larger matrix into smaller ones using the createSubMatrix method given below:

public static Matrix createSubMatrix(Matrix matrix, int excluding_row, int excluding_col) {
    Matrix mat = new Matrix(matrix.getNrows()-1, matrix.getNcols()-1);
    int r = -1;
    for (int i=0;i<matrix.getNrows();i++) {
        if (i==excluding_row)
            continue;
            r++;
            int c = -1;
        for (int j=0;j<matrix.getNcols();j++) {
            if (j==excluding_col)
                continue;
            mat.setValueAt(r, ++c, matrix.getValueAt(i, j));
        }
    }
    return mat;
} 

The input parameters for this method are the original matrix and the row and column index numbers that need to be deleted from the original matrix to create the sub-matrix.

Cofactor of a Matrix

The cofactor of a matrix A is matrix C that the value of element C_ij equals the determinant of a matrix created by removing row i and column j from matrix A. Here is the method that calculates the cofactor matrix:

public static Matrix cofactor(Matrix matrix) throws NoSquareException {
    Matrix mat = new Matrix(matrix.getNrows(), matrix.getNcols());
    for (int i=0;i<matrix.getNrows();i++) {
        for (int j=0; j<matrix.getNcols();j++) {
            mat.setValueAt(i, j, changeSign(i) * changeSign(j) * 
                             determinant(createSubMatrix(matrix, i, j)));
        }
    }
    
    return mat;
}

This method is necessary to calculate the inverse of a matrix given in the next section. For details about cofactor, visit this link.

Inverse of a Matrix

Inverse of a square matrix A is the matrix A^-1 where AA^-1=I. I is the identity matrix (see this link for more details).

public static Matrix inverse(Matrix matrix) throws NoSquareException {
    return (transpose(cofactor(matrix)).multiplyByConstant(1.0/determinant(matrix)));
}

In this method, the inverse of a matrix is calculated by finding the transpose of the cofactor of that matrix divided by the determinant of that matrix. Not all of square matrices have inverse. If the matrix is not invertible (a singular matrix), the value of the matrix coming out of the above method will be NAN (stands for not a number) or Infinity.

For matrix multiplication, addition, and subtraction, see the attached code.

Points of Interest

All of the above operations are fundamental in linear algebra and perhaps the inverse of a matrix is the hardest operation among others to understand and implement.

History

This is the first version (v1.0.2).