2012-03-10 13 views
24

Esta pregunta es la inversa de this question.Conversión de un hash anidado en un hash plano

Dado un resumen anidada como

{ 
    :a => { 
     :b => {:c => 1, :d => 2}, 
     :e => 3, 
    }, 
    :f => 4, 
} 

¿cuál es la mejor manera de convertirlo en un hash plana como

{ 
    [:a, :b, :c] => 1, 
    [:a, :b, :d] => 2, 
    [:a, :e] => 3, 
    [:f] => 4, 
} 
+3

Entonces, ¿quieres una matriz de claves para generar un valor? – Linuxios

Respuesta

15

Otra forma:

def flat_hash(h,f=[],g={}) 
    return g.update({ f=>h }) unless h.is_a? Hash 
    h.each { |k,r| flat_hash(r,f+[k],g) } 
    g 
end 

h = { :a => { :b => { :c => 1, 
         :d => 2 }, 
       :e => 3 }, 
     :f => 4 } 

flat_hash(h) #=> {[:a, :b, :c]=>1, [:a, :b, :d]=>2, [:a, :e]=>3, [:f]=>4} 
+0

Esto es muy rápido. – sawa

+0

Sé que Ruby no necesariamente admite TCO de fábrica, pero si no devolvía 'g' aquí, ¿se optimizaría esto? – rusty

+0

He escuchado el término "cola optimizada" pero no tengo idea de lo que significa. Tal vez un científico de la computación pueda responder su pregunta. –

5

Esto no es un intento de darle la mejor manera a hazlo, pero es una forma: P

def flatten(hash) 
    return {[] => hash} if !hash.is_a?(Hash) 
    map = {} 
    hash.each_pair do |key1, value1| 
    flatten(value1).each_pair do |key2, value2| 
     map[[key1] + key2] = value2 
    end 
    end 
    return map 
end 

Funciona para su ejemplo, produciendo este resultado:

{[:a, :b, :c]=>1, [:a, :b, :d]=>2, [:a, :e]=>3, [:f]=>4} 

Es posible que no se obtengan los resultados esperados si hay valores hash vacíos.

15

muy similar a la solución de Adiel Mittmann

def flat_hash(h, k = []) 
    new_hash = {} 
    h.each_pair do |key, val| 
    if val.is_a?(Hash) 
     new_hash.merge!(flat_hash(val, k + [key])) 
    else 
     new_hash[k + [key]] = val 
    end 
    end 
    new_hash 
end 

Editar: refactorizado para la elegancia. Debería ser casi tan rápido.

def flat_hash(hash, k = []) 
    return {k => hash} unless hash.is_a?(Hash) 
    hash.inject({}){ |h, v| h.merge! flat_hash(v[-1], k + [v[0]]) } 
end 
+0

Este corrió más rápido. Gracias. – sawa

+1

@sawa: solo un consejo para el futuro: si desea una solución rápida, mencione esto en la pregunta la próxima vez. Por lo general, el criterio principal en los lenguajes dinámicos como Python o Ruby es la elegancia y la concisión. Si también pides específicamente el rendimiento, podrías obtener respuestas mucho más adecuadas :) –

+1

@sawa: refactorizado por elegancia. – Kyle

8

Mi intento:

def flatten_hash(h) 
    return { [] => h } unless h.is_a?(Hash) 
    Hash[h.map { |a,v1| flatten_hash(v1).map { |b,v2| [[a] + b, v2] } }.flatten(1)] 
end 

Lo siento por los malos nombres de las variables, tenía que encajar en una sola línea.

3

un enfoque funcional (ver la history durante implementaciones alternativas):

def recursive_flatten(hash) 
    hash.flat_map do |key, value| 
    if value.is_a?(Hash) 
     recursive_flatten(value).map { |ks, v| [[key] + ks, v] } 
    else 
     [[[key], value]] 
    end 
    end.to_h 
end 
1

Inspirado por forma @ Cary-swoveland, pero en la clase Hash:

class Hash 
    def deep_flatten(previous_key=[]) 
    flat_hash = {} 
    self.each do |key, value| 
     next_key = previous_key+[key] 
     flat_hash.update(value.is_a?(Hash) ? value.deep_flatten(next_key) : {next_key=>value}) 
    end 
    return flat_hash 
    end 
end 

h = { :a => { :b => { :c => 1, :d => 2 }, :e => 3 }, :f => 4 } 

h.deep_flatten #=> {[:a, :b, :c]=>1, [:a, :b, :d]=>2, [:a, :e]=>3, [:f]=>4} 
0

Una solución declarativa usando DeepEnumerable:

require 'deep_enumerable' 

h = { :a => { :b => { :c => 1, :d => 2 }, :e => 3 }, :f => 4 } 

h.deep_each.map do |k, v| 
    [DeepEnumerable.deep_key_to_array(k), v] 
end.to_h 

o, para aquellos que prefieren libre de punto estilo

h.deep_each.to_h.shallow_map_keys(&DeepEnumerable.method(:deep_key_to_array)) 
1

soporte para matrices/nombres legibles/no actualización para teclas de velocidad/resultados Stringified

def flat_hash(input, base = nil, all = {}) 
    if input.is_a?(Array) 
    input = input.each_with_index.to_a.each(&:reverse!) 
    end 

    if input.is_a?(Hash) || input.is_a?(Array) 
    input.each do |k, v| 
     flat_hash(v, base ? "#{base}.#{k}" : k, all) 
    end 
    else 
    all[base] = input 
    end 

    all 
end