Ruby - 如何用数组值反转哈希？

Question

寻找适用于 Ruby 1.8.7 的答案：

例如，假设我有一个像这样的哈希：

{"Book Y"=>["author B", "author C"], "Book X"=>["author A", "author B", "author C"]}

我想得到这个：

{ 
    "author A" => ["Book X"],
    "author B" => ["Book Y", "Book X"],
    "author C" => ["Book Y", "Book X"] 
}

我为它编写了一个很长的方法，但是对于大型数据集，它非常慢。

有什么优雅的解决方案吗？

Answer 1

这是一种方法：

g = {"Book Y"=>["author B", "author C"],
     "Book X"=>["author A", "author B", "author C"]}

g.each_with_object({}) do |(book,authors),h|
  authors.each { |author| (h[author] ||= []) << book }
end
  #=> {"author B"=>["Book Y", "Book X"],
  #    "author C"=>["Book Y", "Book X"],
  #    "author A"=>["Book X"]} 

步骤：

enum = g.each_with_object({})
  #=> #<Enumerator: {"Book Y"=>["author B", "author C"],
  #   "Book X"=>["author A", "author B", "author C"]}:each_with_object({})> 

通过将其转换为数组，我们可以看到 的元素enum，它将传递到块中：

enum.to_a
  #=> [[["Book Y", ["author B", "author C"]], {}],
  #    [["Book X", ["author A", "author B", "author C"]], {}]]

enum传递到块并分配给块变量的第一个元素是：

(book,authors),h = enum.next
  #=> [["Book Y", ["author B", "author C"]], {}] 
book
  #=> "Book Y" 
authors
  #=> ["author B", "author C"] 
h
  #=> {} 

enum1 = authors.each
  #=> #<Enumerator: ["author B", "author C"]:each>
author = enum1.next
  #=> "author B"
(h[author] ||= []) << book
  #=> (h["author B"] ||= []) << "Book Y"
  #=> (h["author B"] = h["author B"] || []) << "Book Y"
  #=> (h["author B"] = nil || []) << "Book Y"
  #=> h["author B"] = ["Book Y"]
  #=> ["Book Y"]
h #=> {"author B"=>["Book Y"]} 

下一个：

author = enum1.next
  #=> "author C" 
(h[author] ||= []) << book
h #=> {"author B"=>["Book Y", "Book Y"], "author C"=>["Book Y"]} 

完成后"Book X"，

(book,authors),h = enum.next
  #=> [["Book X", ["author A", "author B", "author C"]],
  #    {"author B"=>["Book Y", "Book Y"], "author C"=>["Book Y"]}]
book
  #=> "Book X" 
authors
  #=> ["author A", "author B", "author C"] 
h
  #=> {"author B"=>["Book Y", "Book Y"], "author C"=>["Book Y"]} 

现在，我们重复与 相同的计算"Book X"。唯一的区别是，当我们遇到：

(h[author] ||= []) << book

这相当于

(h[author] = h[author] || []) << book

在大多数情况h[author]下，等号右边不会是nil（例如，它可能是["Book X"]，在这种情况下，上面的表达式简化为：

h[author] << book

附录

对于战前的 Ruby 版本（例如 1.8.7），只需首先初始化哈希并使用each代替each_with_object（我们在 1.9 中得到了后者。我对 1.8.7 还太年轻，但我经常想知道如果没有 1.8.7，人们是如何相处的）你只需要记住h在最后返回，因为each只是返回它的接收者。

所以将其更改为：

h = {}
g.each do |book,authors|
  authors.each { |author| (h[author] ||= []) << book }
end
h
  #=> {"author B"=>["Book Y", "Book X"],
  #    "author C"=>["Book Y", "Book X"],
  #    "author A"=>["Book X"]}