2015-07-13 11 views
5

Esiste una più semplice trasposizione del seguente hash? Ho una soluzione che funziona ma il metodo di trasposizione è difficile da leggere. Il caso d'uso è un hash di lavori (j1, j2, j3, ecc.) E le date in cui si verificano (d1, d2, d3, ecc ...). L'obbligo è di prendere un hash di eventi (e1, e2, e3 ecc.) Raggruppati per data quindi lavoro e trasferirli agli eventi raggruppati per lavoro quindi data.Trasposizione di un hash in rubino con zip

Ho sperimentato con il metodo #zip ma ciò richiederebbe l'iniezione di nil su j1: d1 e quindi di rimuoverli dai risultati. Ho anche faticato a trovare un esempio di utilizzo di #zip con un argomento di blocco. Capisco che sia #zip with a block always returns nil ma di conseguenza non sono riuscito a capire come mai lo useresti praticamente.

require 'rspec' 
require 'pry' 

#  | d1 | d2 | d3 | 
#  ---------------------- 
# j1 |  | e2 | e3 | 
# -------------------------- 
# j2 | e4 | e5 | e6 | 
# -------------------------- 
# j3 | e7 |  | e9 | 
# -------------------------- 

def transpose(h) 
    Hash[ 
    dates(h).map do |d| 
     [ 
     d, 
     Hash[ h.keys.map do |j| 
      h[j][d] ? [j, h[j][d]] : nil 
     end.compact ] 
     ] 
    end 
    ] 
end 

def dates(h) 
    h.values.map(&:keys).reduce(:|).sort 
end 

describe "transpose" do 

    let(:starting) { 
    { 
     j1: { d2: :e2, d3: :e3 }, 
     j2: { d1: :e4, d2: :e5, d3: :e6 }, 
     j3: { d1: :e7, d3: :e9 } 
    } 
    } 

    let(:transposed) { 
    { 
     d1: { j2: :e4, j3: :e7 }, 
     d2: { j1: :e2, j2: :e5 }, 
     d3: { j1: :e3, j2: :e6, j3: :e9 } 
    } 
    } 

    it { expect(dates(starting)).to eq([:d1, :d2, :d3]) } 
    it { expect(transpose(starting)).to eq(transposed) } 
end 
+2

Vorrei che ogni domanda qui fosse così. Un buon esempio di codice, con test. – Yule

+0

E 'abbastanza difficile seguire la tua descrizione però - penso che stai usando la parola "hash" in posti che intendevi "array", è corretto? ('hash di lavori',' hash di date') - L'esempio alla fine lo cancella comunque. :) – BroiSatse

+0

Grazie a @Yule. Scrivere domande è un'arte in sé! –

risposta

2

Ho riscritto il tuo transpose metodo un po ', che dovrebbe essere più veloce e pulito:

def transpose(h) 
    h.each_with_object({}) do |(outer, data), ret| 
     data.each do |inner, event| 
     ret[inner] = {} unless ret[inner] 
     ret[inner][outer] = event 
     end 
    end 
    end 

Non usa la mappatura non necessario di date, e funziona in entrambe le direzioni (cambia interna ed esterna chiavi). Fatemi sapere cosa ne pensate :)

+0

Bello! Il bello di fare una domanda con i test è che è possibile copiare e incollare il codice per vedere se i test passano :) Questo è molto più pulito della mia soluzione. Grazie. ➜ sd git: (PT98271602_scrollable_calendar) ✗ ./stack_overflow_question.rb rspec .. finito in 0.00393 secondi 2 esempi, 0 fallimenti –