Ruby:获取散列中的所有键(包括子键)

问题描述 投票:0回答:10

让我们有这个哈希:

hash = {"a" => 1, "b" => {"c" => 3}}
hash.get_all_keys 
=> ["a", "b", "c"]

我怎样才能得到所有的钥匙,因为

hash.keys
只返回
["a", "b"]

ruby hash
10个回答
31
投票

这将为您提供任何嵌套级别的所有键的数组。

def get_em(h)
  h.each_with_object([]) do |(k,v),keys|      
    keys << k
    keys.concat(get_em(v)) if v.is_a? Hash
  end
end

hash = {"a" => 1, "b" => {"c" => {"d" => 3}}}
get_em(hash) #  => ["a", "b", "c", "d"]

8
投票

我发现

grep
在这里很有用:

def get_keys(hash)
  ( hash.keys + hash.values.grep(Hash){|sub_hash| get_keys(sub_hash) } ).flatten
end

p get_keys my_nested_hash #=> ["a", "b", "c"]

我喜欢这个解决方案,因为它很短,但读起来非常好。


5
投票

保留键层次结构的版本

  • 使用数组
  • 适用于嵌套哈希

keys_only.rb

# one-liner
def keys_only(h); h.map { |k, v| v = v.first if v.is_a?(Array); v.is_a?(Hash) ? [k, keys_only(v)] : k }; end

# nicer
def keys_only(h)
  h.map do |k, v|
    v = v.first if v.is_a?(Array);

    if v.is_a?(Hash)
      [k, keys_only(v)]
    else
      k
    end
  end
end

hash = { a: 1, b: { c: { d: 3 } }, e: [{ f: 3 }, { f: 5 }] }
keys_only(hash)
# => [:a, [:b, [[:c, [:d]]]], [:e, [:f]]]

P.S.:是的,它看起来像一个词法分析器 :D

奖励:在漂亮的嵌套列表中打印键

# one-liner
def print_keys(a, n = 0); a.each { |el| el.is_a?(Array) ? el[1] && el[1].class == Array ? print_keys(el, n) : print_keys(el, n + 1) : (puts "  " * n + "- #{el}") }; nil; end

# nicer
def print_keys(a, n = 0)
  a.each do |el|
    if el.is_a?(Array)
       if el[1] && el[1].class == Array
         print_keys(el, n)
       else
         print_keys(el, n + 1)
       end
    else
      puts "  " * n + "- #{el}"
    end
  end

  nil
end


> print_keys(keys_only(hash))
- a
  - b
      - c
        - d
  - e
    - f

3
投票
def get_all_keys(hash)
  hash.map do |k, v|
    Hash === v ? [k, get_all_keys(v)] : [k]
  end.flatten
end

2
投票

还处理包含散列的嵌套数组

def all_keys(items)
  case items
  when Hash then items.keys + items.values.flat_map { |v| all_keys(v) }
  when Array then items.flat_map { |i| all_keys(i) }
  else []
  end
end

2
投票

请看下面的代码,只有一个嵌套级别:

hash = {"a" => 1, "b" => {"c" => 3}}
keys = hash.keys + hash.select{|_,value|value.is_a?(Hash)}
       .map{|_,value| value.keys}.flatten
p keys

结果:

["a", "b", "c"]

更新:新的递归解决方案,考虑多级嵌套输入(如@Bala的评论,例如:

hash = {"a" => 1, "b" => {"c" => {"c" => 3}}, "c" => {"b" => 3}}
)。

class Hash
  def recursive_keys
    if any?{|_,value| value.is_a?(Hash)}
       keys + select{|_,value|value.is_a?(Hash)}
                    .map{|_,value| value.recursive_keys}.flatten
    else
       keys
    end
  end
end

hash =  {"a" => 1, "b" => {"c" => {"d" => 3}}, "e" => {"f" => 3}}
p hash.recursive_keys

结果:

["a", "b", "e", "c", "d", "f"]

1
投票
class Hash

  def get_all_keys
    [].tap do |result|
      result << keys
      values.select { |v| v.respond_to?(:get_all_keys) }.each do |value| 
        result << value.get_all_keys
      end
    end.flatten
  end

end

hash = {"a" => 1, "b" => {"c" => 3}}
puts hash.get_all_keys.inspect # => ["a", "b", "c"]

1
投票

这是另一种方法:

def get_all_keys(h)
  h.each_with_object([]){|(k,v),a| v.is_a?(Hash) ? a.push(k,*get_all_keys(v)) : a << k }
end

hash = {"a" => 1, "b" => {"c" => {"d" => 3}}}
p get_all_keys(hash)
# >> ["a", "b", "c", "d"]

0
投票

我确定有更优雅的解决方案,但这个选项有效:

blah = {"a" => 1, "b" => {"c" => 3}}
results = []
blah.each do |k,v|
  if v.is_a? Hash
    results << k
    v.each_key {|key| results << key}
  else
    results << k
  end
end
puts results

0
投票

hash.keys
是我见过的最简单的返回散列中的键值数组的方法。

© www.soinside.com 2019 - 2024. All rights reserved.