Como trocar chaves e valores em um hash


154

Como troco chaves e valores em um Hash?

Eu tenho o seguinte Hash:

{:a=>:one, :b=>:two, :c=>:three}

que eu quero transformar em:

{:one=>:a, :two=>:b, :three=>:c}

Usar mapparece um pouco tedioso. Existe uma solução mais curta?

Respostas:


280

O Ruby possui um método auxiliar para o Hash que permite tratar um Hash como se estivesse invertido (em essência, permitindo acessar as chaves por meio de valores):

{a: 1, b: 2, c: 3}.key(1)
=> :a

Se você deseja manter o hash invertido, o Hash # invert deve funcionar na maioria das situações:

{a: 1, b: 2, c: 3}.invert
=> {1=>:a, 2=>:b, 3=>:c}

MAS...

Se você tiver valores duplicados, invertdescartará tudo, exceto a última ocorrência dos seus valores (porque continuará substituindo o novo valor para essa chave durante a iteração). Da mesma forma, keyretornará apenas a primeira partida:

{a: 1, b: 2, c: 2}.key(2)
=> :b

{a: 1, b: 2, c: 2}.invert
=> {1=>:a, 2=>:c}

Portanto, se seus valores forem únicos, você poderá usar Hash#invert. Caso contrário, você pode manter todos os valores como uma matriz, assim:

class Hash
  # like invert but not lossy
  # {"one"=>1,"two"=>2, "1"=>1, "2"=>2}.inverse => {1=>["one", "1"], 2=>["two", "2"]} 
  def safe_invert
    each_with_object({}) do |(key,value),out| 
      out[value] ||= []
      out[value] << key
    end
  end
end

Nota: Este código com testes está agora no GitHub .

Ou:

class Hash
  def safe_invert
    self.each_with_object({}){|(k,v),o|(o[v]||=[])<<k}
  end
end

4
each_with_objectfaz mais sentido aqui do que inject.
Andrew Marshall

de modo que se torna each_with_object({}){ |i,o|k,v = *i; o[v] ||=[]; o[v] << k}... agradável
Nigel Thorne

3
AMD. eu não sabia que você poderia fazer | (chave, valor), sair |. Isso é tão incrível, eu odiava aquela matriz chegando lá em vez de chave e valor. Muito obrigado
Iuri G.

63

Você apostou que há um! Sempre há uma maneira mais curta de fazer as coisas no Ruby!

É bem simples, basta usar Hash#invert:

{a: :one, b: :two, c: :three}.invert
=> {:one=>:a, :two=>:b, :three=>:c}

Et voilà!


4
Hash # invert não funcionará se os mesmos valores aparecerem várias vezes no seu hash.
Tilo

2
files = {
  'Input.txt' => 'Randy',
  'Code.py' => 'Stan',
  'Output.txt' => 'Randy'
}

h = Hash.new{|h,k| h[k] = []} # Create hash that defaults unknown keys to empty an empty list
files.map {|k,v| h[v]<< k} #append each key to the list at a known value
puts h

Isso também manipulará os valores duplicados.


1
Você pode explicar a resposta que está acontecendo em cada etapa?
Sajjad Murtaza

Pessoalmente, evito definir o comportamento do valor padrão do hash. Eu me preocupo que qualquer código que eu forneça esse hash não espere que um hash se comporte dessa maneira, e isso poderá causar algum erro insidioso mais tarde. Eu realmente não posso justificar essa preocupação. É apenas uma dúvida que não consigo ignorar. Princípio da menor surpresa?
Nigel Thorne

Ao responder com código, é realmente importante explicar como o código funciona e por que é a solução apropriada. O objetivo é educar, não apenas resolver o problema imediato.
the Tin Man

1
# this doesn't looks quite as elegant as the other solutions here,
# but if you call inverse twice, it will preserve the elements of the original hash

# true inversion of Ruby Hash / preserves all elements in original hash
# e.g. hash.inverse.inverse ~ h

class Hash

  def inverse
    i = Hash.new
    self.each_pair{ |k,v|
      if (v.class == Array)
        v.each{ |x|
          i[x] = i.has_key?(x) ? [k,i[x]].flatten : k
        }
      else
        i[v] = i.has_key?(v) ? [k,i[v]].flatten : k
      end
    }
    return i
  end

end

Hash#inverse da-te:

 h = {a: 1, b: 2, c: 2}
 h.inverse
  => {1=>:a, 2=>[:c, :b]}
 h.inverse.inverse
  => {:a=>1, :c=>2, :b=>2}  # order might not be preserved
 h.inverse.inverse == h
  => true                   # true-ish because order might change

enquanto o invertmétodo interno está quebrado:

 h.invert
  => {1=>:a, 2=>:c}    # FAIL
 h.invert.invert == h 
  => false             # FAIL

1

Usando matriz

input = {:key1=>"value1", :key2=>"value2", :key3=>"value3", :key4=>"value4", :key5=>"value5"}
output = Hash[input.to_a.map{|m| m.reverse}]

Usando Hash

input = {:key1=>"value1", :key2=>"value2", :key3=>"value3", :key4=>"value4", :key5=>"value5"}
output = input.invert

1

Se você tiver um hash, onde as chaves são únicas, use Hash # invert :

> {a: 1, b: 2, c: 3}.invert
=> {1=>:a, 2=>:b, 3=>:c} 

Isso não funcionará se você tiver chaves não exclusivas, no entanto, onde apenas as últimas chaves vistas serão mantidas:

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

Se você tiver um hash com chaves não exclusivas, poderá fazer:

> hash={a: 1, b: 2, c: 3, d: 3, e: 2, f: 1}
> hash.each_with_object(Hash.new { |h,k| h[k]=[] }) {|(k,v), h| 
            h[v] << k
            }     
=> {1=>[:a, :f], 2=>[:b, :e], 3=>[:c, :d]}

Se os valores do hash já forem matrizes, você poderá:

> hash={ "A" => [14, 15, 16], "B" => [17, 15], "C" => [35, 15] }
> hash.each_with_object(Hash.new { |h,k| h[k]=[] }) {|(k,v), h| 
            v.map {|t| h[t] << k}
            }   
=> {14=>["A"], 15=>["A", "B", "C"], 16=>["A"], 17=>["B"], 35=>["C"]}
Ao utilizar nosso site, você reconhece que leu e compreendeu nossa Política de Cookies e nossa Política de Privacidade.
Licensed under cc by-sa 3.0 with attribution required.