7

我正在尝试根据各个值在某个范围内的位置将整数数组分组到哈希中。基本上我想将数组转换为固定宽度的直方图。

例子:

values = [1,3,4,4,4,4,4,10,12,15,18]
bin_width = 3

我需要将数组值分组到一个基于范围的直方图中,它们落入一个 3 单位宽的桶中,如下所示:

{'0..2'=>[1,3],'3..5'=>[4,4,4,4,4],'6..8'=>[],'9..11'=>[10]....

是否有一个简单的单行解决方案(可能类似于values.group_by{|x| #range calc})可以在这里工作?

4

2 回答 2

11
values = [1, 7, 2, 8, 2]
values.group_by { |x| x / 3 }.map { |k, vs| [(3*k..3*k+2), vs] }.to_h
#=> {0..2=>[1, 2, 2], 6..8=>[7, 8]}

如果您真的需要空范围,我认为不可能有一个干净的单线。但这应该这样做:

grouped = values.group_by { |x| x / 3 }
min, max = grouped.keys.minmax
(min..max).map { |n| [(3*n..3*n+2), grouped.fetch(n, [])] }.to_h
#=> {0..2=>[1, 2, 2], 3..5=>[], 6..8=>[7, 8]}
于 2012-07-31T14:09:25.040 回答
4

我想出了一个效率很低但很清晰的解决方案:

ranges = 0.step(values.max, bin_width).each_cons(2).map { |s, e| Range.new(s, e, true) }
values.group_by { |v| ranges.find { |r| r.cover? v } }
于 2012-07-31T14:34:47.527 回答