ruby - 如何在 ruby​​ 中找到第一个值的数组之间的并集

Question

我有一组这样的数组:

[["1","2"],["1","3"],["2","3"],["2","5"]]

我想找到像这样的第一个值的并集

["1","2"],["1","3"] 匹配，所以我需要创建新数组，如 ["1","2,3 "]

所以结果数组会像

[["1","2,3"],["2","3,5"]]

Answer 1

与 Ruby 中的大多数问题一样，Enumerable 模块可以完成以下工作:

input = [["1","2"],["1","3"],["2","3"],["2","5"]]

result = input.group_by do |item|
  # Group by first element
  item[0]
end.collect do |key, items|
  # Compose into new format
  [
    key,
    items.collect do |item|
      item[1]
    end.join(',')
  ]
end

puts result.inspect
# => [["1", "2,3"], ["2", "3,5"]]

group_by 方法在像这样聚合时非常方便，而 collect 非常适合重写元素的显示方式。