如何将两个文本文件与ruby中的值进行比较

Question

我有两个文本文件，它们有一个相同的字段（宠物）。我想合并两个文件并输出一个包含三个字段的新文件（所有者宠物购买者）。我的代码主要取决于两个文件是否具有相同的宠物（宠物的名称和数量），而不是我将买方名称添加到新文件中的买方字段中。只有当两个文件具有相同的pet文件时才应该打印出来。

但是我的代码没有按我的意愿运行，需要一些帮助，谢谢。

input_file_1 (.txt)

owner pet
Michael dog, cat
John pig, rabbit
Marry dog, cat

input_file_2 (.txt)

buyer pet
Sean cat, dog
Mark cat, dog
Joy dog, mouse
Tina cat, dog

我希望结果如下所示：

owner pet buyer
Michael cat, dog Sean, Mark, Tina
Mary cat, dog Sean, Mark, Tina

我的代码如下所示：

input_file_1 = ARGV[0]
input_file_2 = ARGV[1]

hash_1 = {}
File.readlines(input_file_1, "\n").each do |line|
  owner, pet = line.chomp.split("\t")
  hash_1[owner] = pet
end

hash_2 = {}
File.readlines(input_file_2, "\n").each do |line|
  buyer, pet = line.chomp.split("\t")
  hash_2[buyer] = pet
end

hash_1.each do |key, value|
  if hash_2.has_value? value
    puts "#{key}\t#{value}\t#{hash_2[key]}"
  end
end

Answer 1

我建议你使用pet 作为键：

input_file_1 = ARGV[0]
input_file_2 = ARGV[1]

hash_1 = Hash.new([])
File.readlines(input_file_1, "\n").each do |line|
  owner, pet = line.chomp.split("\t")
  hash_1[pet] += [owner]
end

hash_2 = Hash.new([])
File.readlines(input_file_2, "\n").each do |line|
  buyer, pet = line.chomp.split("\t")
  hash_2[pet] += [buyer]
end

hash_1.each do |pet, owners|
  if hash_2.include? pet
    owners.each do |owner|
      puts "#{owner}\t#{pet}\t#{hash_2[pet].join(", ")}"
    end
  end
end