将多个数据文件合并到一个输出中，然后从中创建一个文件

Question

早上，

我正在尝试在 rails 中创建一个函数来合并 3 个文件， 我有 2 个 CSV 文件和 1 个 JSON 文件，这些文件都与书籍有关， 这 3 个文件具有相同书籍的不同部分数据， 所以一个 CSV 有一个标题和 ID，另一个有一个不同的标题，具有相同的 ID 和一个 DOI，JSON 文件有一个以作者姓名作为键的哈希，值是书籍的所有 DOI 的数组。

示例数据：

CSV 一：

DOI,Title,ID
1111,Ruby on Rails,1234-1234
2222,Rails 4.0,4321-4321

CSV 二：

Title,ID
Ruby,1234-1234
Rails,4321-4321

JSON：

[{"name":"Homer Simpson","articles":["1111","2222"]}]

示例输出（作为 JSON），

    [{
    "doi": "1111",
    "title": "Ruby on Rails",
    "author": "Homer Simpson",
    "book": "Ruby",
    "ID": "1234-1234"
    }
    {
    "doi": "2222",
    "title": "Rails 4.0",
    "author": "Homer Simpson",
    "book": "Rails",
    "ID": "4321-4321"
    }]

我知道我首先需要解析数据，将 CSV 转换为数组，然后遍历它们，我正在努力解决的问题是合并到单个哈希中。

Answer 1

require 'csv'
require 'json'

csv1 = CSV.open('csv1.csv').read
csv2 = CSV.open('csv2.csv').read
csv1.shift
csv2.shift
@json = JSON.parse('[{"name":"Homer Simpson","articles":["1111","2222"]}]')

@csv1_data = csv1.inject([]) do |res, el|
  res << { doi: el[0], title: el[1], id: el[2] }
  res
end

@csv2_data = csv2.inject([]) do |res, el|
  res << { book: el[0], id: el[1] }
  res
end

def book_attributes(id)
  @csv2_data.find { |el| el[:id] == id }
end

def author_attributes(doi)
  item = @json.find { |el| el['articles'].include? doi }
  { author: item['name'] }
end


result = @csv1_data.inject([]) do |res, el|
  details = el.merge(author_attributes(el[:doi]))
  details.merge!(book_attributes(el[:id]))
  res << details
  res
end

puts result

希望这样做。