Расширение Array.prototype возвращает undefined

Question

Я пытаюсь расширить Array.prototype, чтобы включить квадратную функцию. У меня есть это:

Array.prototype.square = function(){ 
    return this.forEach(function(el){ 
    return (el * el) 
    }); 
} 

Когда я вызываю эту функцию на массив, скажем, arr = [2, 2, 2] он возвращает неопределенное значение. Если я добавлю console.log, я вижу, что функция обратного вызова для функции forEach выполняется правильно - она ​​регистрируется 4 раза три раза. Почему эта функция возвращает undefined вместо нового массива из [4, 4, 4]?

Answer 1

Метод forEach не возвращает значение. Вы должны использовать map:

Array.prototype.square = function(){ 
    return this.map(function(el){ 
    return (el * el) 
    }); 
} 

console.log([2, 2, 2].square()); // [4, 4, 4] 

Answer 2

Как p.s.w.g. сказал: .map - это подходящая функция, но в комментарии вы спросили об использовании forEach. Для того, чтобы получить эту работу, вы должны создать временный массив:

Array.prototype.square = function(){ 
    var tmp = []; 

    this.forEach(function(el){ 
    tmp.push(el * el) 
    }); 

    return tmp; 
} 

console.log([2, 2, 2].square()); // [4, 4, 4] 

.map() лучше, хотя.