Question

require 'csv'    

CSV.foreach(filename, :headers => true) do |row|
  if column3 = true || column 4 = true
    Model.create!(row.to_hash)
  else
    skip
  end
end

首先，有没有办法在row.to_hash期间只抓取某些列？

其次，我使用if语句是获取特定行的最佳方法吗？我可以将整个CSV加载到各种各样的临时表中，然后抓住我需要的东西吗？

Answer 1

row.to_hash将根据标头生成属性哈希。要选择一组特定的属性，您应该使用切片。

>> row.to_hash # { :attr1 => 'val1', :attr2 => 'val2', :attr3 => 'val3' }
>> row.to_hash.slice(:attr1, :attr2) # { :attr1 => 'val1', :attr2 => 'val2' }

关于你的第二个问题：是的，你仍然需要加载整个csv，然后检查每一行。

使用rails从CSV导入特定的行和列

1 个答案: