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

Question

执行以下操作的“红宝石”方式是什么？我仍在考虑更多的命令式编程，并没有真正适应 ruby​​ 的思考。我想要做的是找到大小最接近数组平均值的元素，例如，考虑以下数组

[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]
从该数组中选择第一个 results 个条目。

Answer 2

我假设需要的是数组中小于平均值的最大元素和大于平均值的数组最小值。当且仅当数组具有至少两个元素并且它们不完全相同时，这样的值才存在。假设条件适用，我们只需将其从单词转换为符号：

avg = a.reduce(:+)/a.size.to_f
[ a.select { |e| e < avg }.max, a.select { |e| e > avg }.min ]

另一种方式，效率稍低：

avg = a.reduce(:+)/a.size.to_f
b = (a + [avg]).uniq.sort
i = b.index(avg)
[ b[i-1], b[i+1] ]