js封装一个函数，输入一个数组和一个布尔，数组包括5嗝浮点数字，当布尔为false时，返回求和值，当布尔值为true时，返回平均值

以下是一个js函数，输入一个数组和一个布尔，数组包括5个浮点数字，当布尔为false时，返回求和值，当布尔值为true时，返回平均值：

function calculate(arr, bool) {
if (bool) {
var sum = 0;
for (var i = 0; i < arr.length; i++) {
sum += arr[i];
}
return sum / arr.length;
} else {
var sum = 0;
for (var i = 0; i < arr.length; i++) {
sum += arr[i];
}
return sum;
}
}