#!/usr/bin/env ruby lines = File.readlines(ARGV[0]) lines.shift mut = File.readlines(ARGV[1]) mut.shift mutagenicity = {} mut.each do |l| smi,m = l.chomp.split(",") if m == "1" mutagenicity[smi] = "mutagen" elsif m == "0" mutagenicity[smi] = "non-mutagen" else mutagenicity[smi] = "NA" end end lines.collect{|l| l.chomp.split(",").first}.each do |smi| smi.gsub!('"','') mutagenicity[smi] ? puts(mutagenicity[smi]) : puts("PA") end