Як витягти підхеш із хешу?


95

У мене є хеш:

h1 = {:a => :A, :b => :B, :c => :C, :d => :D}

Який найкращий спосіб витягти такий підхеш?

h1.extract_subhash(:b, :d, :e, :f) # => {:b => :B, :d => :D}
h1 #=> {:a => :A, :c => :C}


можливий дублікат фільтра Ruby Hash
Джон Дворак

1
@JanDvorak Це питання стосується не лише повернення підхеша, а й модифікації існуючого. Дуже схожі речі, але ActiveSupport має різні засоби боротьби з ними.
skalee

Відповіді:


58

Якщо ви спеціально хочете, щоб метод повернув витягнуті елементи, але h1 залишився незмінним:

h1 = {:a => :A, :b => :B, :c => :C, :d => :D}
h2 = h1.select {|key, value| [:b, :d, :e, :f].include?(key) } # => {:b=>:B, :d=>:D} 
h1 = Hash[h1.to_a - h2.to_a] # => {:a=>:A, :c=>:C} 

І якщо ви хочете внести це в клас Hash:

class Hash
  def extract_subhash(*extract)
    h2 = self.select{|key, value| extract.include?(key) }
    self.delete_if {|key, value| extract.include?(key) }
    h2
  end
end

Якщо ви просто хочете видалити зазначені елементи з хешу, це набагато простіше за допомогою delete_if .

h1 = {:a => :A, :b => :B, :c => :C, :d => :D}
h1.delete_if {|key, value| [:b, :d, :e, :f].include?(key) } # => {:a=>:A, :c=>:C} 
h1  # => {:a=>:A, :c=>:C} 

2
Це O (n2) - у вас буде один цикл на select, інший цикл на include, який буде називатися h1.size times.
metakungfu

1
Хоча ця відповідь гідна для чистого рубіну, якщо ви використовуєте рейки, відповідь нижче (з використанням вбудованих sliceабо except, залежно від ваших потреб) набагато чистіший
Krease

137

ActiveSupport, Принаймні , так як 2.3.8, забезпечує чотири зручні методи: #slice, #exceptі їх руйнівні аналоги: #slice!і #except!. Вони згадувалися в інших відповідях, але підсумовуючи їх в одному місці:

x = {a: 1, b: 2, c: 3, d: 4}
# => {:a=>1, :b=>2, :c=>3, :d=>4}

x.slice(:a, :b)
# => {:a=>1, :b=>2}

x
# => {:a=>1, :b=>2, :c=>3, :d=>4}

x.except(:a, :b)
# => {:c=>3, :d=>4}

x
# => {:a=>1, :b=>2, :c=>3, :d=>4}

Зверніть увагу на значення повернення методів вибуху. Вони не лише адаптують наявний хеш, але й повернуть видалені (не зберігаються) записи. У Hash#except!костюмах кращий приклад , наведений в цьому питанні:

x = {a: 1, b: 2, c: 3, d: 4}
# => {:a=>1, :b=>2, :c=>3, :d=>4}

x.except!(:c, :d)
# => {:a=>1, :b=>2}

x
# => {:a=>1, :b=>2}

ActiveSupportне вимагає цілих перил, досить легкий. Насправді, від цього залежить багато дорогоцінних каменів, які, можливо, вже є у Gemfile.lock. Не потрібно самостійно поширювати клас Hash.


3
Результат x.except!(:c, :d)(з ударом) повинен бути # => {:a=>1, :b=>2}. Добре, якщо ви зможете відредагувати свою відповідь.
244 квітня

28

Якщо ви використовуєте рейки , хеш-шматочок - це шлях.

{:a => :A, :b => :B, :c => :C, :d => :D}.slice(:a, :c)
# =>  {:a => :A, :c => :C}

Якщо ви не використовуєте рейки , Hash # values_at поверне значення в тому ж порядку, як ви їх запитали, щоб ви могли це зробити:

def slice(hash, *keys)
  Hash[ [keys, hash.values_at(*keys)].transpose]
end

def except(hash, *keys)
  desired_keys = hash.keys - keys
  Hash[ [desired_keys, hash.values_at(*desired_keys)].transpose]
end

напр .:

slice({foo: 'bar', 'bar' => 'foo', 2 => 'two'}, 'bar', 2) 
# => {'bar' => 'foo', 2 => 'two'}

except({foo: 'bar', 'bar' => 'foo', 2 => 'two'}, 'bar', 2) 
# => {:foo => 'bar'}

Пояснення:

З {:a => 1, :b => 2, :c => 3}ми хочемо{:a => 1, :b => 2}

hash = {:a => 1, :b => 2, :c => 3}
keys = [:a, :b]
values = hash.values_at(*keys) #=> [1, 2]
transposed_matrix =[keys, values].transpose #=> [[:a, 1], [:b, 2]]
Hash[transposed_matrix] #=> {:a => 1, :b => 2}

Якщо вам здається, що це маніпулювання виправленням мавп, це наступне:

module MyExtension
  module Hash 
    def slice(*keys)
      ::Hash[[keys, self.values_at(*keys)].transpose]
    end
    def except(*keys)
      desired_keys = self.keys - keys
      ::Hash[[desired_keys, self.values_at(*desired_keys)].transpose]
    end
  end
end
Hash.include MyExtension::Hash

2
Виправлення Mokey - це, безумовно, шлях до IMO. Набагато чистіший і робить намір чіткішим.
Ромаріо

1
Додайте, щоб змінити код для адреси основного модуля, визначте модуль та імпортуйте розширити ядро ​​хешу ... модуль Модуль CoreExtensions Хеш розріз (* ключі) :: Хеш [[ключі, self.values_at (* ключі)]. Транспонування] кінець кінець end Hash.include CoreExtensions :: Hash
Ронан Фоглас


5

Ви можете використовувати slice! (* Клавіші), який доступний у основних розширеннях ActiveSupport

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

extracted_slice = initial_hash.slice!(:a, :c)

Initial_hash тепер буде

{:b => 2, :d =>4}

extracted_slide тепер буде

{:a => 1, :c =>3}

Можна подивитися slice.rb in ActiveSupport 3.1.3


Думаю, ви описуєте витяг !. екстракт! видаляє ключі з початкового хешу, повертаючи новий хеш, що містить видалені ключі. скибочку! робить навпаки: видаляє всі, крім зазначених ключів, з початкового хешу (знову повертає новий хеш, що містить видалені ключі). Тож скибочку! трохи більше схожа на операцію "збереження".
Russ Egan

1
ActiveSupport не є частиною Ruby STI
Volte

4
module HashExtensions
  def subhash(*keys)
    keys = keys.select { |k| key?(k) }
    Hash[keys.zip(values_at(*keys))]
  end
end

Hash.send(:include, HashExtensions)

{:a => :A, :b => :B, :c => :C, :d => :D}.subhash(:a) # => {:a => :A}

1
Хороша робота. Не зовсім те, про що він просить. Ваш метод повертає: {: d =>: D,: b =>: B,: e => nil,: f => nil} {: c =>: C,: a =>: A,: d => : D,: b =>: B}
Енді

Еквівалентне однорядкове (і, можливо, швидше) рішення: <pre> def subhash(*keys) select {|k,v| keys.include?(k)} end
пік

3
h1 = {:a => :A, :b => :B, :c => :C, :d => :D}
keys = [:b, :d, :e, :f]

h2 = (h1.keys & keys).each_with_object({}) { |k,h| h.update(k=>h1.delete(k)) }
  #=> {:b => :B, :d => :D}
h1
  #=> {:a => :A, :c => :C}

2

якщо ви використовуєте рейки, може бути зручно використовувати Hash.except

h = {a:1, b:2}
h1 = h.except(:a) # {b:2}

1
class Hash
  def extract(*keys)
    key_index = Hash[keys.map{ |k| [k, true] }] # depends on the size of keys
    partition{ |k, v| key_index.has_key?(k) }.map{ |group| Hash[group] }  
  end
end

h1 = {:a => :A, :b => :B, :c => :C, :d => :D}
h2, h1 = h1.extract(:b, :d, :e, :f)

1

Ось швидке порівняння продуктивності запропонованих методів, #selectздається, найшвидшим

k = 1_000_000
Benchmark.bmbm do |x|
  x.report('select') { k.times { {a: 1, b: 2, c: 3}.select { |k, _v| [:a, :b].include?(k) } } }
  x.report('hash transpose') { k.times { Hash[ [[:a, :b], {a: 1, b: 2, c: 3}.fetch_values(:a, :b)].transpose ] } }
  x.report('slice') { k.times { {a: 1, b: 2, c: 3}.slice(:a, :b) } }
end

Rehearsal --------------------------------------------------
select           1.640000   0.010000   1.650000 (  1.651426)
hash transpose   1.720000   0.010000   1.730000 (  1.729950)
slice            1.740000   0.010000   1.750000 (  1.748204)
----------------------------------------- total: 5.130000sec

                     user     system      total        real
select           1.670000   0.010000   1.680000 (  1.683415)
hash transpose   1.680000   0.010000   1.690000 (  1.688110)
slice            1.800000   0.010000   1.810000 (  1.816215)

Доопрацювання буде виглядати так:

module CoreExtensions
  module Extractable
    refine Hash do
      def extract(*keys)
        select { |k, _v| keys.include?(k) }
      end
    end
  end
end

І щоб використовувати його:

using ::CoreExtensions::Extractable
{ a: 1, b: 2, c: 3 }.extract(:a, :b)

1

І те, delete_ifі інше keep_ifє частиною ядра Ruby. Тут ви можете досягти того, що хотіли б, не виправляючи Hashтип.

h1 = {:a => :A, :b => :B, :c => :C, :d => :D}
h2 = h1.clone
p h1.keep_if { |key| [:b, :d, :e, :f].include?(key) } # => {:b => :B, :d => :D}
p h2.delete_if { |key, value| [:b, :d, :e, :f].include?(key) } #=> {:a => :A, :c => :C}

Для отримання додаткової інформації перевірте посилання нижче з документації:


1

Як згадували інші, Ruby 2.5 додав метод зрізу Hash #.

Rails 5.2.0beta1 також додав власну версію Hash # slice, щоб розширити функціональність для користувачів фреймворку, які використовують попередню версію Ruby. https://github.com/rails/rails/commit/01ae39660243bc5f0a986e20f9c9bff312b1b5f8

Якщо ви хочете реалізувати свій власний з будь-якої причини, це також хороший вкладиш:

 def slice(*keys)
   keys.each_with_object(Hash.new) { |k, hash| hash[k] = self[k] if has_key?(k) }
 end unless method_defined?(:slice)

0

Цей код вводить функціональність, яку ви просите, у клас Hash:

class Hash
    def extract_subhash! *keys
      to_keep = self.keys.to_a - keys
      to_delete = Hash[self.select{|k,v| !to_keep.include? k}]
      self.delete_if {|k,v| !to_keep.include? k}
      to_delete
    end
end

і дає результати, які ви надали:

h1 = {:a => :A, :b => :B, :c => :C, :d => :D}
p h1.extract_subhash!(:b, :d, :e, :f) # => {b => :B, :d => :D}
p h1 #=> {:a => :A, :c => :C}

Примітка: цей метод фактично повертає витягнуті ключі / значення.


0

Ось функціональне рішення, яке може бути корисним, якщо ви не працюєте на Ruby 2.5 і якщо ви не хочете забруднити свій клас Hash, додавши новий метод:

slice_hash = -> keys, hash { hash.select { |k, _v| keys.include?(k) } }.curry

Тоді ви можете застосувати його навіть до вкладених хешів:

my_hash = [{name: "Joe", age: 34}, {name: "Amy", age: 55}]
my_hash.map(&slice_hash.([:name]))
# => [{:name=>"Joe"}, {:name=>"Amy"}]

0

Просто доповнення до методу зрізу, якщо клавіші підмета, які ви хочете відокремити від оригінального хешу, будуть динамічними, ви можете зробити, наприклад,

slice(*dynamic_keys) # dynamic_keys should be an array type 

0

Ми можемо зробити це, зациклювавши лише ключі, які ми хочемо витягти, і просто перевіривши, чи існує ключ, а потім витягти його.

class Hash
  def extract(*keys)
    extracted_hash = {}
    keys.each{|key| extracted_hash[key] = self.delete(key) if self.has_key?(key)}
    extracted_hash
  end
end
h1 = {:a => :A, :b => :B, :c => :C, :d => :D}
h2 = h1.extract(:b, :d, :e, :f)
Використовуючи наш веб-сайт, ви визнаєте, що прочитали та зрозуміли наші Політику щодо файлів cookie та Політику конфіденційності.
Licensed under cc by-sa 3.0 with attribution required.