Question

我的Array包含Hash，其中只有一个键和值为Array。

例如：

a = [{1 => ["foo", "bar"]}, {2 => ["hello"]}, {3 => ["world", "bar"]}]

现在，我想创建一个Hash，其值为Hash以上为密钥，其键为值。

例如:(期望的结果）

res =  {"foo"=>[1], "bar"=>[1, 3], "hello"=>[2], "world"=>[3]}

我已经解决了以下问题：

b =  Hash.new { |h, k| h[k] = [] }
a.each { |hash|
  hash.each { |key, vals|
    vals.each { |val|
      b[val] << key
    }
  }
}
b
# => {"foo"=>[1], "bar"=>[1, 3], "hello"=>[2], "world"=>[3]}

它运行正常，但应该有一个更好，更短的方法来做这个迭代这么多次。请建议。

Answer 1

可以反转键和值，然后根据需要收集元素：

a = [{1 => ["foo", "bar"]}, {2 => ["hello"]}, {3 => ["world", "bar"]}]

a.map(&:invert).inject({}) { |memo,el| 
    el = el.flatten   # hash of size 1 ⇒ array [k ⇒ v]
    el.first.each { |k| (memo[k] ||= []) << el.last }
    memo 
}

#⇒ {
#  "bar" => [
#    [0] 1,
#    [1] 3
#  ],
#  "foo" => [
#    [0] 1
#  ],
#  "hello" => [
#    [0] 2
#  ],
#  "world" => [
#    [0] 3
#  ]
# }

希望它有所帮助。

Answer 2

我使用group_by，但您需要按摩数据：

a.flat_map(&:to_a) # => [[1, ["foo", "bar"]], [2, ["hello"]], [3, ["world", "bar"]]]
.flat_map{|key, values| values.map{|v| [key, v]}} # => [[1, "foo"], [1, "bar"], [2, "hello"], [3, "world"], [3, "bar"]]
.group_by(&:last) # => {"foo"=>[[1, "foo"]], "bar"=>[[1, "bar"], [3, "bar"]], "hello"=>[[2, "hello"]], "world"=>[[3, "world"]]}
.map{|key, values| [key, values.map(&:first)]} # =>  [["foo", [1]], ["bar", [1, 3]], ["hello", [2]], ["world", [3]]]
.to_h # => {"foo"=>[1], "bar"=>[1, 3], "hello"=>[2], "world"=>[3]}

将哈希数组反向映射到哈希

2 个答案: