ruby - 将嵌套散列展平为数组

Question

我想将嵌套哈希展平为数组。例如：

a = {'1'=>{'2'=>{'5'=>{},'6'=>{'8'=>{}}}},'3'=>{},'4'=>{'7'=>{}}}

flatten_nested_hash(a) 的结果是：

["1", "2", "5", "6", "8", "3", "4", "7"]

最后我写了一些递归函数，但我觉得必须有一些更简单，非递归的方式来做。

我的功能如下所示：

  def flatten_nested_hash(categories)
    categories.map do |k,v|
      if v == {} 
        k
      else
        [k,flatten_nested_hash(v)]
      end
    end.flatten
  end

Answer 1

递归的。

def flatten_nested_hash(categories)
  categories.flat_map{|k, v| [k, *flatten_nested_hash(v)]}
end

在 Hash 类上定义它。

class Hash
  def flatten_nested; flat_map{|k, v| [k, *v.flatten_nested]} end
end

Answer 2

使用 Ruby 2.1 及更高版本，您可以使用改进来添加方法Hash并将其仅公开给您需要的模块/类。

module HashRefinements
  refine Hash do
    def flatten_nested
      flat_map { |k, v| [k, *v.flatten_nested] }
    end
  end
end

在您的班级/模块中

class MyThing
  using HashRefinements

  ...

  def flatten_categories
    categories.flatten_nested
  end

  ...
end

Answer 3

这是一个嵌套的数据结构——您将不得不使用某种递归或迭代方法来提取所有键。不过，这比您拥有的要容易一些：

def deep_extract_keys(hash)
  hash.keys + hash.values.flat_map {|value| deep_extract_keys value }
end

p deep_extract_keys({"1"=>{"2"=>{"5"=>{}, "6"=>{"8"=>{}}}}, "3"=>{}, "4"=>{"7"=>{}}})

这是广度优先搜索，而不是深度优先搜索，因此输出为：

["1", "3", "4", "2", "5", "6", "8", "7"]

Answer 4

这不是（显式）递归的，仅适用于非负整数键：

a.to_s.scan(/\d+/).map(&:to_i)   # [1, 2, 5, 6, 8, 3, 4, 7]

:-)

Answer 5

module Flattener
  def deep_flatten
    flatten.map do |item|
      case item
      when Hash, Array
        item.deep_flatten
      else
        item
      end
    end.flatten
  end
end

class Hash

  include Flattener

end

class Array

  include Flattener

end