web-dev-qa-db-ja.com

JavaScriptでの2D配列の転置

私は次のような配列の配列を持っています：

[
    [1,2,3],
    [1,2,3],
    [1,2,3],
]

それを転置して、次の配列を取得したいと思います。

[
    [1,1,1],
    [2,2,2],
    [3,3,3],
]

ループを使用してプログラムで行うことは難しくありません。

function transposeArray(array, arrayLength){
    var newArray = [];
    for(var i = 0; i < array.length; i++){
        newArray.Push([]);
    };

    for(var i = 0; i < array.length; i++){
        for(var j = 0; j < arrayLength; j++){
            newArray[j].Push(array[i][j]);
        };
    };

    return newArray;
}

しかし、これはかさばるようなもので、もっと簡単な方法があるはずです。ある？

javascriptarraysmatrixtranspose
128
ckersch
array[0].map((col, i) => array.map(row => row[i]));

mapは、配列内の要素ごとに提供されたcallback関数を順番に1回呼び出し、結果から新しい配列を作成します。 callbackは、値が割り当てられた配列のインデックスに対してのみ呼び出されます。削除されたインデックスまたは値が割り当てられたことのないインデックスに対しては呼び出されません。

callbackは、要素の値、要素のインデックス、および走査されるArrayオブジェクトの3つの引数で呼び出されます。 [ソース]

163
Fawad Ghafoor

nderscore.js を使用できます

_.Zip.apply(_, [[1,2,3], [1,2,3], [1,2,3]])
38
Joe

ここに最新のブラウザでの実装があります（依存関係なし）：

transpose = m => m[0].map((x,i) => m.map(x => x[i]))
32
Mahdi Jadaliha

lodash/underscoreおよびes6を使用した最短の方法：

_.Zip(...matrix)

matrixは次のとおりです。

const matrix = [[1,2,3], [1,2,3], [1,2,3]];
20
marcel

ここで多くの良い答え！それらを1つの回答に統合し、いくつかのコードを最新の構文に更新しました。

Fawad Ghafoor および ÓscarGómezAlcañizに触発されたワンライナー

function transpose(matrix) {
  return matrix[0].map((col, i) => matrix.map(row => row[i]));
}

function transpose(matrix) {
  return matrix[0].map((col, c) => matrix.map((row, r) => matrix[r][c]));
}

Andrew Tatomyrによるreduceを使用した機能的アプローチスタイル

function transpose(matrix) {
  return matrix.reduce((prev, next) => next.map((item, i) =>
    (prev[i] || []).concat(next[i])
  ), []);
}

marcelによるダッシュ/アンダースコア

function tranpose(matrix) {
  return _.Zip(...matrix);
}

// Without spread operator.
function transpose(matrix) {
  return _.Zip.apply(_, [[1,2,3], [1,2,3], [1,2,3]])
}

バニラアプローチ

function transpose(matrix) {
  const rows = matrix.length, cols = matrix[0].length;
  const grid = [];
  for (let j = 0; j < cols; j++) {
    grid[j] = Array(rows);
  }
  for (let i = 0; i < rows; i++) {
    for (let j = 0; j < cols; j++) {
      grid[j][i] = matrix[i][j];
    }
  }
  return grid;
}

Emanuel Saringanに触発されたバニラインプレースES6アプローチ

function transpose(matrix) {
  for (var i = 0; i < matrix.length; i++) {
    for (var j = 0; j < i; j++) {
      const temp = matrix[i][j];
      matrix[i][j] = matrix[j][i];
      matrix[j][i] = temp;
    }
  }
}

// Using destructing
function transpose(matrix) {
  for (var i = 0; i < matrix.length; i++) {
    for (var j = 0; j < i; j++) {
      [matrix[i][j], matrix[j][i]] = [matrix[j][i], matrix[i][j]];
    }
  }
}
12
Yangshun Tay

きちんとした、純粋な：

[[0, 1], [2, 3], [4, 5]].reduce((prev, next) => next.map((item, i) =>
    (prev[i] || []).concat(next[i])
), []); // [[0, 2, 4], [1, 3, 5]]

空の配列が提供されている場合、以前のソリューションでは失敗する可能性があります。

ここに関数としてあります：

function transpose(array) {
    return array.reduce((prev, next) => next.map((item, i) =>
        (prev[i] || []).concat(next[i])
    ), []);
}

console.log(transpose([[0, 1], [2, 3], [4, 5]]));

更新スプレッド演算子を使用するとさらに改善できます。

const transpose = matrix => matrix.reduce(($, row) =>
    row.map((_, i) => [...($[i] || []), row[i]]), 
    []
)
9
Andrew Tatomyr

1つのパスのみを実行することで、インプレースで実行できます。

function transpose(arr,arrLen) {
  for (var i = 0; i < arrLen; i++) {
    for (var j = 0; j <i; j++) {
      //swap element[i,j] and element[j,i]
      var temp = arr[i][j];
      arr[i][j] = arr[j][i];
      arr[j][i] = temp;
    }
  }
}
9
Emanuel Saringan

Array.mapを使用した別のバリエーション。インデックスを使用すると、M != Nの行列を転置できます。

// Get just the first row to iterate columns first
var t = matrix[0].map(function (col, c) {
    // For each column, iterate all rows
    return matrix.map(function (row, r) { 
        return matrix[r][c]; 
    }); 
});

転置するために必要なことは、最初に要素を列ごとに、次に行ごとにマッピングすることです。

6
Óscar Gómez Alcañiz

RamdaJS を使用することがオプションの場合、これは1行で実現できます：R.transpose(myArray)

4
Rafael Rozon

Ramda JSおよびES6構文を使用するオプションがある場合、これを行う別の方法があります。

const transpose = a => R.map(c => R.map(r => r[c], a), R.keys(a[0]));

console.log(transpose([
  [1, 2, 3, 4],
  [5, 6, 7, 8],
  [9, 10, 11, 12]
])); // =>  [[1,5,9],[2,6,10],[3,7,11],[4,8,12]]
<script src="https://cdnjs.cloudflare.com/ajax/libs/ramda/0.22.1/ramda.min.js"></script>
4
Kevin Le - Khnle

配列を外側から内側に繰り返し、内側の値をマッピングして行列を減らす別のアプローチ。

const
    transpose = array => array.reduce((r, a) => a.map((v, i) => [...(r[i] || []), v]), []),
    matrix = [[1, 2, 3], [1, 2, 3], [1, 2, 3]];

console.log(transpose(matrix));
3
Nina Scholz

編集：この答えは行列を転置するのではなく、回転させます。最初の質問を注意深く読んでいませんでした：D

時計回りおよび反時計回りの回転：

    function rotateCounterClockwise(a){
        var n=a.length;
        for (var i=0; i<n/2; i++) {
            for (var j=i; j<n-i-1; j++) {
                var tmp=a[i][j];
                a[i][j]=a[j][n-i-1];
                a[j][n-i-1]=a[n-i-1][n-j-1];
                a[n-i-1][n-j-1]=a[n-j-1][i];
                a[n-j-1][i]=tmp;
            }
        }
        return a;
    }

    function rotateClockwise(a) {
        var n=a.length;
        for (var i=0; i<n/2; i++) {
            for (var j=i; j<n-i-1; j++) {
                var tmp=a[i][j];
                a[i][j]=a[n-j-1][i];
                a[n-j-1][i]=a[n-i-1][n-j-1];
                a[n-i-1][n-j-1]=a[j][n-i-1];
                a[j][n-i-1]=tmp;
            }
        }
        return a;
    }
3
Aryan Firouzian

ES6 1ライナー：

let invert = a => a[0].map((col, c) => a.map((row, r) => a[r][c]))

óscarのものと同じですが、時計回りに回転させる方がいいでしょう：

let rotate = a => a[0].map((col, c) => a.map((row, r) => a[r][c]).reverse())
2
ysle

以下を使用して、ループなしでこれを実現できます。

非常にエレガントに見え、 jQuery of nderscore.js のような依存関係を必要としません。

function transpose(matrix) {  
    return zeroFill(getMatrixWidth(matrix)).map(function(r, i) {
        return zeroFill(matrix.length).map(function(c, j) {
            return matrix[j][i];
        });
    });
}

function getMatrixWidth(matrix) {
    return matrix.reduce(function (result, row) {
        return Math.max(result, row.length);
    }, 0);
}

function zeroFill(n) {
    return new Array(n+1).join('0').split('').map(Number);
}

縮小

function transpose(m){return zeroFill(m.reduce(function(m,r){return Math.max(m,r.length)},0)).map(function(r,i){return zeroFill(m.length).map(function(c,j){return m[j][i]})})}function zeroFill(n){return new Array(n+1).join("0").split("").map(Number)}

ここに私が一緒に投げたデモがあります。ループがないことに注意してください:-)

// Create a 5 row, by 9 column matrix.
var m = CoordinateMatrix(5, 9);

// Make the matrix an irregular shape.
m[2] = m[2].slice(0, 5);
m[4].pop();

// Transpose and print the matrix.
println(formatMatrix(transpose(m)));

function Matrix(rows, cols, defaultVal) {
    return AbstractMatrix(rows, cols, function(r, i) {
        return arrayFill(cols, defaultVal);
    });
}
function ZeroMatrix(rows, cols) {
    return AbstractMatrix(rows, cols, function(r, i) {
        return zeroFill(cols);
    });
}
function CoordinateMatrix(rows, cols) {
    return AbstractMatrix(rows, cols, function(r, i) {
        return zeroFill(cols).map(function(c, j) {
            return [i, j];
        });
    });
}
function AbstractMatrix(rows, cols, rowFn) {
    return zeroFill(rows).map(function(r, i) {
        return rowFn(r, i);
    });
}
/** Matrix functions. */
function formatMatrix(matrix) {
    return matrix.reduce(function (result, row) {
        return result + row.join('\t') + '\n';
    }, '');
}
function copy(matrix) {  
    return zeroFill(matrix.length).map(function(r, i) {
        return zeroFill(getMatrixWidth(matrix)).map(function(c, j) {
            return matrix[i][j];
        });
    });
}
function transpose(matrix) {  
    return zeroFill(getMatrixWidth(matrix)).map(function(r, i) {
        return zeroFill(matrix.length).map(function(c, j) {
            return matrix[j][i];
        });
    });
}
function getMatrixWidth(matrix) {
    return matrix.reduce(function (result, row) {
        return Math.max(result, row.length);
    }, 0);
}
/** Array fill functions. */
function zeroFill(n) {
  return new Array(n+1).join('0').split('').map(Number);
}
function arrayFill(n, defaultValue) {
    return zeroFill(n).map(function(value) {
        return defaultValue || value;
    });
}
/** Print functions. */
function print(str) {
    str = Array.isArray(str) ? str.join(' ') : str;
    return document.getElementById('out').innerHTML += str || '';
}
function println(str) {
    print.call(null, [].slice.call(arguments, 0).concat(['<br />']));
}
#out {
    white-space: pre;
}
<div id="out"></div>
2
Mr. Polywhirl

これは少し読みやすいと思います。 Array.from を使用し、ロジックはネストされたループを使用するのと同じです。

var arr = [
  [1, 2, 3, 4],
  [1, 2, 3, 4],
  [1, 2, 3, 4]
];

/*
 * arr[0].length = 4 = number of result rows
 * arr.length = 3 = number of result cols
 */

var result = Array.from({ length: arr[0].length }, function(x, row) {
  return Array.from({ length: arr.length }, function(x, col) {
    return arr[col][row];
  });
});

console.log(result);

長さが等しくない配列を扱う場合は、arr[0].lengthを別のものに置き換える必要があります。

var arr = [
  [1, 2],
  [1, 2, 3],
  [1, 2, 3, 4]
];

/*
 * arr[0].length = 4 = number of result rows
 * arr.length = 3 = number of result cols
 */

var result = Array.from({ length: arr.reduce(function(max, item) { return item.length > max ? item.length : max; }, 0) }, function(x, row) {
  return Array.from({ length: arr.length }, function(x, col) {
    return arr[col][row];
  });
});

console.log(result);
1
Salman A

上記の答えは読みにくいか冗長すぎると感じたので、自分で書きます。そして、これは線形代数で転置を実装する最も直感的な方法だと思います、あなたは値交換を行わず、正しい要素を各要素に挿入するだけです新しいマトリックス：

function transpose(matrix) {
  const rows = matrix.length
  const cols = matrix[0].length

  let grid = []
  for (let col = 0; col < cols; col++) {
    grid[col] = []
  }
  for (let row = 0; row < rows; row++) {
    for (let col = 0; col < cols; col++) {
      grid[col][row] = matrix[row][col]
    }
  }
  return grid
}
1
Chang
function invertArray(array,arrayWidth,arrayHeight) {
  var newArray = [];
  for (x=0;x<arrayWidth;x++) {
    newArray[x] = [];
    for (y=0;y<arrayHeight;y++) {
        newArray[x][y] = array[y][x];
    }
  }
  return newArray;
}
0
Samuel Reid

配列を切り捨てないマトリックスシェイプで動作する、TypeScriptのライブラリフリーの実装：

const rotate2dArray = <T>(array2d: T[][]) => {
    const rotated2d: T[][] = []

    return array2d.reduce((acc, array1d, index2d) => {
        array1d.forEach((value, index1d) => {
            if (!acc[index1d]) acc[index1d] = []

            acc[index1d][index2d] = value
        })

        return acc
    }, rotated2d)
}
0
pirix-gh