javascriptで行列

var mat = [];

for(var i=0;i<4;i++){
    mat.push([]);
    for(var j=0;j<4;j++){
       mat[i].push(0);
    }
}

for(var k=0;k<mat.length;k++){
 if(k==0){
   mat[k][k] = 1;
 }
 if(k==1){
   mat[k][k] = 1;
 }
 if(k==2){
   mat[k][k] = 1;
 }
 if(k==3){
   mat[k][k] = 1;
 }
}
mat;

最初↑を書いたけど、

function mmmat(n){
  var mat = [];

  for(var i=0;i<n;i++){
     mat.push([]);
     for(var j=0;j<n;j++){
        mat[i].push(0);
     }
  }

  for(var k=0;k<mat.length;k++){
      mat[k][k] = 1;
  }
  return mat;
}
alert(mmmat(4));

↑と書けると気が付いた。
if文いらないですね。