web-dev-qa-db-ja.com

ハッシュの配列をフィルタリングして、別の配列のキーのみを取得するにはどうすればよいですか？

配列内の各ハッシュのキーのサブセットを取得しようとしています。

ハッシュは実際にははるかに大きくなりますが、私はこれが理解しやすいと考えました：

[
  {
    id:2,
    start: "3:30",
    break: 30,
    num_attendees: 14
  },
  {
    id: 3,
    start: "3: 40",
    break: 40,
    num_attendees: 4
  },
  {
    id: 4,
    start: "4: 40",
    break: 10,
    num_attendees: 40
  }
]

idとstartの値のみを取得したい。

私はもう試した：

return_keys = ['id','start']
return_array = events.select{|key,val|  key.to_s.in? return_keys}

しかし、これは空の配列を返します。

rubyarrayshash
29
pedalpete

これはあなたが望むことをするはずです：

events.map do |hash|
  hash.select do |key, value|
    [:id, :start].include? key
  end
end

潜在的に高速な（しかし、やや劣る）ソリューション：

events.map do |hash|
  { id: hash[:id], start: hash[:start] }
end

必要な場合return_keys動的にする：

return_keys = [:id, :start]
events.map do |hash|
  {}.tap do |new_hash|
    return_keys.each do |key|
      new_hash[key] = hash[key]
    end
  end
end

コードでは、selectはarrayから要素を選択します。配列内に含まれるハッシュを変更します。

パフォーマンスが心配な場合は、ここにリストされているすべてのソリューションのベンチマークを行いました（ code ）：

                user     system      total        real
amarshall 1  0.140000   0.000000   0.140000 (  0.140316)
amarshall 2  0.060000   0.000000   0.060000 (  0.066409)
amarshall 3  0.100000   0.000000   0.100000 (  0.101469)
tadman 1     0.140000   0.010000   0.150000 (  0.145489)
tadman 2     0.110000   0.000000   0.110000 (  0.111838)
mu           0.130000   0.000000   0.130000 (  0.128688)
44
Andrew Marshall

Rails（またはActiveSupportの全部または一部を引っ張っても構いません）を使用している場合は、 Hash#slice ：

return_array = events.map { |h| h.slice(:id, :start) }

Hash#slice はカバーの下で追加の作業を行いますが、おそらく十分に高速なので、小さなハッシュでは気付かないでしょうし、明快さはかなりいいです。

32
mu is too short

より良い解決策は、キーごとに線形配列検索を行う代わりに、インデックスとしてハッシュを使用することです。

events = [{id:2, start:"3:30",break:30,num_attendees:14},{id:3, start:"3:40",break:40,num_attendees:4},{id:4, start:"4:40",break:10,num_attendees:40}]

return_keys = [ :id, :start ]

# Compute a quick hash to extract the right values: { key => true }
key_index = Hash[return_keys.collect { |key| [ key, true ] }]

return_array = events.collect do |event|
  event.select do |key, value|
    key_index[key]
  end
end

# => [{:id=>2, :start=>"3:30"}, {:id=>3, :start=>"3:40"}, {:id=>4, :start=>"4:40"}]

eventsの定義に一致するキー名としてシンボルを使用するように調整しました。

これは、return_keysを直接ドライバーとして使用することでさらに改善できます。

events = [{id:2, start:"3:30",break:30,num_attendees:14},{id:3, start:"3:40",break:40,num_attendees:4},{id:4, start:"4:40",break:10,num_attendees:40}]

return_keys = [ :id, :start ]

return_array = events.collect do |event|
  Hash[
    return_keys.collect do |key|
      [ key, event[key] ]
    end
  ]
end

結果は同じです。抽出するサブセットが元のサブセットよりもはるかに小さくなる傾向がある場合、これが最善のアプローチかもしれません。

2
tadman

効率が懸念されるように思われることを考慮して、次のことを提案します。

コード

require 'set'

def keep_keys(arr, keeper_keys)
  keepers = keeper_keys.to_set
  arr.map { |h| h.select { |k,_| keepers.include?(k) } }
end

これは Hash＃select を使用します。これは Enumerable＃select とは異なり、ハッシュを返します。 keeper_keysを高速ルックアップ用のセットに変換しました。

例

arr = [{ id:2, start: "3:30", break: 30 },
       { id: 3, break: 40, num_attendees: 4 },
       { break: 10, num_attendees: 40 }]

keep_keys arr, [:id, :start]
  #=> [{:id=>2, :start=>"3:30"}, {:id=>3}, {}] 
keep_keys arr, [:start, :break]
  #=> [{:start=>"3:30", :break=>30}, {:break=>40}, {:break=>10}] 
keep_keys arr, [:id, :start, :cat]
  #=> [{:id=>2, :start=>"3:30"}, {:id=>3}, {}] 
keep_keys arr, [:start]
  #=> [{:start=>"3:30"}, {}, {}] 
keep_keys arr, [:cat, :dog]
0
Cary Swoveland