解析 CSV 文件，将标题字段作为每行的属性

Question

我想解析 CSV 文件，以便将每一行视为一个对象，其中标题行是对象中属性的名称。我可以写这个，但我确信它已经存在了。

这是我的 CSV 输入：

"foo","bar","baz"
1,2,3
"blah",7,"blam"
4,5,6

代码看起来像这样：

CSV.open('my_file.csv','r') do |csv_obj|
  puts csv_obj.foo   #prints 1 the 1st time, "blah" 2nd time, etc
  puts csv.bar       #prints 2 the first time, 7 the 2nd time, etc
end

使用 Ruby 的 CSV 模块，我相信我只能通过索引访问字段。我认为上面的代码会更具可读性。有什么想法吗？

Answer 1

使用 Ruby 1.9 及更高版本，您可以获得一个可索引对象：

CSV.foreach('my_file.csv', :headers => true) do |row|
  puts row['foo'] # prints 1 the 1st time, "blah" 2nd time, etc
  puts row['bar'] # prints 2 the first time, 7 the 2nd time, etc
end

它不是点语法，但它比数字索引更好用。

顺便说一句，对于 Ruby 1.8.x，您需要使用上述语法 FasterCSV 。

Answer 2

下面是使用 Ruby 1.9 的符号语法的示例。在下面的示例中，代码从 Rails db 目录读取名为 data.csv 的 CSV 文件。

<代码>：标题=> true 将第一行视为标题而不是数据行。 <代码>：header_converters => :symbolize 参数然后将标题行中的每个单元格转换为 Ruby 符号。

CSV.foreach("#{Rails.root}/db/data.csv", {:headers => true, :header_converters => :symbol}) do |row|
  puts "#{row[:foo]},#{row[:bar]},#{row[:baz]}"
end

在 Ruby 1.8 中：

require 'fastercsv'
CSV.foreach("#{Rails.root}/db/data.csv", {:headers => true, :header_converters => :symbol}) do |row|
  puts "#{row[:foo]},#{row[:bar]},#{row[:baz]}"
end

根据 Poul（StackOverflow 提问者）提供的 CSV，上面示例代码的输出将是：

1,2,3
blah,7,blam
4,5,6

根据 CSV 文件标题中使用的字符，可能需要将标题输出为为了查看 CSV (FasterCSV) 如何将字符串标题转换为符号。您可以从 CSV.foreach 中输出标头数组。

row.headers

Answer 3

在 Ruby 2.3 中轻松获取哈希：

CSV.foreach('my_file.csv', headers: true, header_converters: :symbol) do |row|
  puts row.to_h[:foo]
  puts row.to_h[:bar]
end

Answer 4

虽然我很晚才开始讨论，但几个月前我在 https:// github.com/vicentereig/virgola。

给定您的 CSV 内容，将它们映射到 FooBar 对象数组非常简单：

"foo","bar","baz"
1,2,3
"blah",7,"blam"
4,5,6

require 'virgola'

class FooBar
  include Virgola

  attribute :foo
  attribute :bar
  attribute :baz
end

csv = <<CSV
"foo","bar","baz"
1,2,3
"blah",7,"blam"
4,5,6
CSV

foo_bars = FooBar.parse(csv).all
foo_bars.each { |foo_bar| puts foo_bar.foo, foo_bar.bar, foo_bar.baz }

Answer 5

因为我经常遇到这个问题：

array_of_hashmaps = CSV.read("path/to/file.csv", headers: true)
puts array_of_hashmaps.first["foo"] # 1

当您想要读取整个文件时，这是非块版本。