ruby - 从 Ruby 中的排序数组创建嵌套哈希——递归 group_by

Question

我有一组对象，这些对象已根据这些对象的几个属性进行了排序。按优先级顺序，这些属性foo是bar和baz。这意味着对象首先按foo;排序 然后具有相同foo值的子序列按bar; 然后那些具有相同foo和bar值的按 排序baz。

我想将其转换为反映此分组的嵌套哈希。基本上我正在寻找一个递归的Enumerable#group_by. 键是foo、bar和baz;的值 这些值可以是对象的子哈希或数组。这是一个例子：

[obj1, obj2, ... objn].group_by_recursive(:foo, :bar, :baz)
#=> {
      foo_val_1 => {
        bar_val_1 => {
          baz_val_1 => [
            obj1,
            obj2,
            obj3
          ],
          baz_val_2 => [
            obj4,
            obj5
          ]
        },
        bar_val_2 => {
          baz_val_1 => [
            obj6,
            obj7
          ],
          baz_val_2 => [
            obj8
          ]
        },
      },
      foo_val_2 => {
        ...
      },
      ...
    }

Answer 1

想出了一个很好的解决方案。像这样的猴子补丁Enumerable：

module Enumerable

  def group_by_recursive(*props)
    groups = group_by(&props.first)
    if props.count == 1
      groups
    else
      groups.merge(groups) do |group, elements|
        elements.group_by_recursive(*props.drop(1))
      end
    end
  end

end

您传递的属性可以是Procs或Symbols

Answer 2

类似于 Sean 的，缺乏错误处理......

class Array
  def nested_group_by(*keys)
    return self if keys.length == 0
    groups = group_by(&keys.shift)
    Hash[groups.map { | k, v | [k, v.nested_group_by(*keys)] }]
  end
end