将简单数组转换为二维数组(矩阵)

将简单数组转换为二维数组(矩阵)

问题描述:

假设我有一个数组:

A = Array(1, 2, 3, 4, 5, 6, 7, 8, 9);

我想把它转换成二维数组(N x M的矩阵),例如这样:

And I want it to convert into 2-dimensional array (matrix of N x M), for instance like this:

A = Array(Array(1, 2, 3), Array(4, 5, 6), Array(7, 8, 9));

请注意,矩阵的行和列是可变的.

Note, that rows and columns of the matrix is changeable.

类似的事情?

function listToMatrix(list, elementsPerSubArray) {
    var matrix = [], i, k;

    for (i = 0, k = -1; i < list.length; i++) {
        if (i % elementsPerSubArray === 0) {
            k++;
            matrix[k] = [];
        }

        matrix[k].push(list[i]);
    }

    return matrix;
}

用法:

var matrix = listToMatrix([1, 2, 3, 4, 4, 5, 6, 7, 8, 9], 3);
// result: [[1, 2, 3], [4, 5, 6], [7, 8, 9]]