извлечение определенного значения из многомерного хэша в ruby ​​по имени ключа - PullRequest
0 голосов
/ 10 февраля 2010

скажем, у меня есть многомерный хеш, а в одной из подшашек у меня есть пара ключ => значение, которую мне нужно получить по ключу. как я могу это сделать?

пример хэшей:

h={:x=>1,:y=>2,:z=>{:a=>{:k=>"needle"}}}
h={:k=>"needle"}

ключ всегда: k, и мне нужно получить «иглу»

я заметил, что в ruby ​​1.8 нет функции "сглаживания" для хэшей, но если бы она была там, я думаю, я бы просто сделал

h.flatten[:k]

я думаю, мне нужно написать рекурсивную функцию для этого?

спасибо

Ответы [ 2 ]

9 голосов
/ 10 февраля 2010

Вы всегда можете написать свое собственное расширение для миссии для Hash, которое сделает за вас грязную работу:

class Hash
  def recursive_find_by_key(key)
    # Create a stack of hashes to search through for the needle which
    # is initially this hash
    stack = [ self ]

    # So long as there are more haystacks to search...
    while (to_search = stack.pop)
      # ...keep searching for this particular key...
      to_search.each do |k, v|
        # ...and return the corresponding value if it is found.
        return v if (k == key)

        # If this value can be recursively searched...
        if (v.respond_to?(:recursive_find_by_key))
          # ...push that on to the list of places to search.
          stack << v
        end
      end
    end
  end
end

Вы можете использовать это довольно просто:

h={:x=>1,:y=>2,:z=>{:a=>{:k=>"needle"}}}

puts h.recursive_find_by_key(:k).inspect
# => "needle"

h={:k=>"needle"}

puts h.recursive_find_by_key(:k).inspect
# => "needle"

puts h.recursive_find_by_key(:foo).inspect
# => nil
2 голосов
/ 13 ноября 2013

Если вам нужно просто получить значение ключа, но вы не знаете, насколько глубоко ключ, используйте этот фрагмент

def find_tag_val(hash, tag)
  hash.map do |k, v|
    return v if k.to_sym == tag
    vr = find_tag_val(v, tag) if v.kind_of?(Hash)
    return vr if vr
  end
  nil #othervice
end 

h = {message: { key: 'val'}}
find_tag_val(h, :key) #=> 'val'
...