查找最接近数组平均值的元素

Question

做以下事情的"红宝石"方式是什么？我仍然在考虑更多命令式的风格编程,而不是真正适应红宝石的思考.我想要做的是找到与数组平均值最接近的元素,例如,考虑以下数组

[1,2,3] 

平均值是2.0.我想写的方法返回最接近平均值的元素,在这种情况下是1和3.

另一个例子将更好地说明这一点

[10,20,50,33,22] avg is 27.0 method would return 22 and 33.

Answer 1

这不是最有效的,但它(以我的拙见)相当于Ruby风格.

class Array
  # Return the single element in the array closest to the average value
  def closest_to_average
    avg = inject(0.0,:+) / length
    min_by{ |v| (v-avg).abs }
  end
end

[1,2,3].closest_to_average
#=> 2 

[10,20,50,33,22].closest_to_average
#=> 22 

如果你真的想要n个最近的项目,那么:

class Array
  # Return a number of elements in the array closest to the average value
  def closest_to_average(results=1)
    avg = inject(0.0,:+) / length
    sort_by{ |v| (v-avg).abs }[0,results]
  end
end

[10,20,50,33,22].closest_to_average     #=> [22] 
[10,20,50,33,22].closest_to_average(2)  #=> [22, 33] 
[10,20,50,33,22].closest_to_average(3)  #=> [22, 33, 20] 

这是怎么回事

avg = inject(0.0,:+) / length
是简写:
avg = self.inject(0.0){ |sum,n| sum+n } / self.length
我开始使用值0.0而不是0确保总和将是一个浮点数,因此除以长度不会给我一个整数舍入值.

sort_by{ |v| (v-avg).abs }
根据数字和平均值(从最低到最高)之间的差异对数组进行排序,然后:从该数组中
[0,results]
选择条目的第一个结果数.