使用Ruby查找X文件中的公共行

时间:2010-08-04 17:32:31

标签: ruby matching

现在我有3个文件,但我希望以后可以添加更多文件的方式执行此操作。每个文件都是ID列表,如下所示。

174535945 174538045 160515924 81712260 25241494

我希望输出是列表1和列表2,列表2和列表3中出现的项目,以及列表1和列表2以及列表3中出现的项目。

最红宝石的方法是创建一个带有每个列表的键的哈希,然后获取所有键并测试所有哈希值,还是有一个很好的宝石可以帮助解决这个问题?

谢谢,

2 个答案:

答案 0 :(得分:4)

使用套装:

require 'set'

list_1 = open(filename_1).read.split.to_set
list_2 = open(filename_2).read.split.to_set
list_3 = open(filename_3).read.split.to_set

puts list_1 & list_2
puts list_2 & list_3
puts list_1 & list_2 & list_3

答案 1 :(得分:0)

我认为您使用哈希的建议是要走的路:

require 'set'

class Test
  def initialize()
    @lists = {174=>[1,2,3], 111=>[2,3,4], 160=>[2,4,5]}
  end

  def common_members(*keys)
    lists_as_sets = {}
    @lists.each_pair {|key,list| lists_as_sets[key] = list.to_set}
    common = lists_as_sets[keys.shift] # Set to key[0] and nil that element of keys
    keys.each {|k| common = common & lists_as_sets[k]} # Intersect the remaining keys, if any
    common.to_a
  end
end

t = Test.new
p t.common_members(174,111) # => [2,3]
p t.common_members(111,160) # => [2,4]
p t.common_members(174,111,160) # =>[2]