RubyでCSVファイルをパースする方法

CSV（Comma-Separated Values）ファイルは、データを表形式で保存するための非常に一般的なフォーマットです。Rubyでは、CSVファイルを簡単に読み書きするための便利なライブラリが用意されています。このガイドでは、Rubyを使用してCSVファイルをパースする方法について詳しく説明します。

CSVライブラリのインポート

まず最初に、Rubyの標準ライブラリに含まれているCSVライブラリをインポートする必要があります。これにより、CSVファイルを操作するためのメソッドが利用可能になります。

require 'csv'

CSVファイルの読み込み

CSVファイルを読み込むには、CSVライブラリの`CSV.foreach`メソッドを使用します。このメソッドは、指定したCSVファイルを行ごとに読み込み、ブロック内で処理を行うことができます。

基本的な読み込み

以下は、CSVファイルを読み込む基本的な例です。

CSV.foreach("data.csv") do |row|
  puts row.inspect
end

このコードは、`data.csv`というファイルを行ごとに読み込み、各行をコンソールに表示します。`row`は配列として各行のデータを保持します。

ヘッダー付きのCSVファイルの読み込み

CSVファイルにヘッダーが含まれている場合、`headers: true`オプションを指定することで、ヘッダーをキーとして持つハッシュとして各行を読み込むことができます。

CSV.foreach("data_with_header.csv", headers: true) do |row|
  puts row['column_name']  # ヘッダー名を指定して値を取得
end

この例では、`data_with_header.csv`というファイルから、特定のカラム名に基づいて値を取得しています。

CSVファイルの書き込み

CSVファイルにデータを書き込むには、`CSV.open`メソッドを使用します。このメソッドは、指定したファイルにデータを追加または新規作成します。

基本的な書き込み

以下は、CSVファイルにデータを書き込む基本的な例です。

CSV.open("output.csv", "w") do |csv|
  csv << ["Name", "Age", "City"]  # ヘッダーの追加
  csv << ["Alice", 30, "Tokyo"]
  csv << ["Bob", 25, "Osaka"]
end

このコードは、`output.csv`という新しいファイルを作成し、ヘッダーと2つのデータ行を書き込みます。

既存のCSVファイルへの追記

既存のCSVファイルにデータを追記するには、`"a"`モードを使用します。

CSV.open("output.csv", "a") do |csv|
  csv << ["Charlie", 28, "Kyoto"]
end

このコードは、`output.csv`ファイルの末尾に新しい行を追加します。

CSVデータの操作

CSVデータを操作するためのいくつかの便利なメソッドを紹介します。

データのフィルタリング

CSVデータをフィルタリングするには、`select`メソッドを使用します。以下の例では、特定の条件に基づいてデータをフィルタリングしています。

filtered_data = CSV.read("data.csv").select { |row| row[1].to_i > 25 }
filtered_data.each { |row| puts row.inspect }

このコードは、`data.csv`ファイルから年齢が25歳以上の行を選択し、表示します。

データのソート

CSVデータをソートするには、`sort_by`メソッドを使用します。以下の例では、年齢でデータをソートしています。

sorted_data = CSV.read("data.csv").sort_by { |row| row[1].to_i }
sorted_data.each { |row| puts row.inspect }

このコードは、`data.csv`ファイルのデータを年齢で昇順にソートし、表示します。

エラーハンドリング

CSVファイルを操作する際には、エラーハンドリングが重要です。ファイルが存在しない場合や、フォーマットが正しくない場合に備えて、例外処理を行うことができます。

begin
  CSV.foreach("non_existent_file.csv") do |row|
    puts row.inspect
  end
rescue Errno::ENOENT
  puts "ファイルが見つかりません。"
rescue CSV::MalformedCSVError
  puts "CSVファイルのフォーマットが正しくありません。"
end

このコードは、指定したファイルが存在しない場合や、CSVフォーマットが不正な場合にエラーメッセージを表示します。

まとめ

Rubyを使用してCSVファイルをパースする方法について学びました。CSVライブラリを使うことで、データの読み込み、書き込み、フィルタリング、ソートが簡単に行えます。エラーハンドリングを適切に行うことで、より堅牢なアプリケーションを作成することができます。

これらの基本を理解することで、CSVファイルを効果的に操作し、データ処理の効率を向上させることができるでしょう。ぜひ、実際のプロジェクトで試してみてください！