csvファイルのよく使う扱い方をまとめました。
CSVファイルを新たに作成
require 'csv'
CSV.open('file.csv', 'w') do |csv|
csv << ["id", "name", "location"]
csv << [1, "hoge_1", "Kyoto"]
# ...
end
file.csv
id,name,location
1,hoge_1,Kyoto
2,hoge_2,Osaka
3,hoge_3,Shiga
4,hoge_4,Fukui
CSVファイルの内容を追記
CSV.open('file.csv', 'a') do |csv|
csv << [5, "hoge_5", "Nagano"]
end
file.csv
id,name,location
1,hoge_1,Kyoto
2,hoge_2,Osaka
3,hoge_3,Shiga
4,hoge_4,Fukui
5,hoge_5,Nagano
CSVファイルを読み込む
配列として扱う場合
CSV.read('file.csv')
# [["id", "name", "location"], ["1", "hoge_1", "Kyoto"] ... ]
ハッシュとして扱う場合（ http://qiita.com/hkengo/items/59ba599ef48c613f2402 ）
CSV.read('file.csv', headers: true)
#<CSV::Table mode:col_or_row row_count:5>
ハッシュからCSVファイル生成例
require 'csv'
module CSVConvertible
def to_csv(*keys)
keys = self.map(&:keys).inject([], &:|) if keys.empty?
CSV.generate do |csv|
csv << keys
self.each { |hash| csv << hash.values_at(*keys) }
end
end
end
users = [
{id: 1, name: 'hoge_1', location: 'Kyoto'},
{id: 2, name: 'hoge_2', location: 'Osaka'},
{id: 3, name: 'hoge_3', location: 'Shiga'},
{id: 4, name: 'hoge_4', location: 'Fukui'}
].extend(CSVConvertible)
IO.write 'file.csv', users.to_csv
file.csv
id,name,location
1,hoge_1,Kyoto
2,hoge_2,Osaka
3,hoge_3,Shiga
4,hoge_4,Fukui
Railsの場合は、ActiveRecordのメソッドを使う。
http://qiita.com/retosu/items/5bda7bd5a55c20fc89c0
参考リンク
https://docs.ruby-lang.org/ja/latest/class/CSV.html
http://qiita.com/hkengo/items/59ba599ef48c613f2402
http://qiita.com/retosu/items/5bda7bd5a55c20fc89c0
http://qiita.com/metheglin/items/439d1fd9b0e0e53c6051