5分でできる！簡単なcsvデータの加工(read/write)法。(Java編)
仕事やプライベートで？受け取ったcsvのデータを、自分の思うようにちょっと加工したいときってありませんか？
項目の順番を入れ替えたり、文字列に加工したりして、自分の使いやすい形に整形したいというシーンは結構あると思います。
Script系の言語を使ったり、SedやAwkを使う！という人もいるでしょうね。私はjavaが使い慣れているので、今回はJavaを使ってみたいと思います。
いくつか便利なライブラリがあるようですが、私は、opencsvが使いやすいと思いました。
opencsvでは、csvファイルの読み込みの方法をいくつか指定することができます。
どの方法を利用して読み込みを行うかはMappingStrategyクラスを切り替えることで指定できます。
HeaderColumnNameTranslateMappingStrategy
HeaderColumnNameMappingStrategy
などはヘッダ項目をもつcsvファイルを扱うのに便利なクラスのようですね。私が今回利用したのはColumnPositionMappingStrategyクラスです。
csvファイルを簡単に加工するというのが目的ですので、読み込み/書き込みの記述が比較的シンプルな方法を選択しました。
準備作業
参照するjar(ライブラリ)
コンパイルに必要なライブラリは以下の通りです。OpenCVはもちろん必要ですが、「commons-beanutils」と「org.apache.commons」も必要なので、詳細は、ここのdependenciesを確認しましょう。
※2016/09/01に再検証した時点では、以下のjarが必要でした。
- opencsv-3.8.jar
- commons-beanutils-1.9.2.jar
- commons-lang3-3.4.jar
読み込みに使うcsvファイル
実際に読み書きに使用するcsvファイルを適当なディレクトリに用意します。サンプルとして、「SampleIn.csv」というファイルをUTF-8の文字コードで作成しました。
|
1 2 3 |
key,name,pref,job key1,name1,pref1,job1 key2,名前2,県2,お仕事2 |
上記のような内容のcsvで良いと思います。1行目がヘッダで、このヘッダ名をキーに下記の「String[]{“key”, “name”, “pref”, “job”}」で対応付けて読み取ります。
csvを扱うクラスの作成
まず、読み込み対象のcsvファイルの項目に対応する単純なJavaBeansクラスを定義します。項目さえ定義すれば、SetterとGetterはEclipseなどのIDEが自動で吐いてくれますよね。
コンストラクタで空白をセットしておいてあげると、空行とかを読み込んだときにエラーで落ちることを防いでくれます。
SampleCsvBeanクラス
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 |
package jp.koji.bean; public class SampleCsvBean { String key; String name; String pref; String job; public SampleCsvBean() { this.key = ""; this.name = ""; this.pref = ""; this.job = ""; } public String getKey() { return key; } public void setKey(String key) { this.key = key; } public String getName() { return name; } public void setName(String name) { this.name = name; } public String getPref() { return pref; } public void setPref(String pref) { this.pref = pref; } public String getJob() { return job; } public void setJob(String job) { this.job = job; } } |
次に、作成したJavaBeansクラスを読み書きする、Mainクラスを作成します。
MainCSVUtil
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 |
package jp.koji.bean; import java.io.FileOutputStream; import java.io.FileInputStream; import java.io.OutputStreamWriter; import java.io.InputStreamReader; import java.io.Reader; import java.io.Writer; import java.util.Iterator; import java.util.List; import com.opencsv.CSVReader; import com.opencsv.CSVWriter; import com.opencsv.bean.ColumnPositionMappingStrategy; import com.opencsv.bean.CsvToBean; public class MainCSVUtil { public static void main(String[] arg) throws Exception { writeSampleCSV("D:\\SampleOut.csv",readSampleCSV("D:\\SampleIn.csv")); } public static List readSampleCSV(String inputFilePath){ Reader ioreader = null; List tmpNDCsvBean = null; try { ioreader = new InputStreamReader(new FileInputStream(inputFilePath),"UTF-8"); ColumnPositionMappingStrategy mappingStrategy = new ColumnPositionMappingStrategy(); mappingStrategy.setType(SampleCsvBean.class); String[] columns = new String[]{"key", "name", "pref", "job"}; mappingStrategy.setColumnMapping(columns); CsvToBean csv = new CsvToBean(); CSVReader reader = new CSVReader(ioreader, ',', '"'); tmpNDCsvBean = csv.parse(mappingStrategy, reader); } catch (Exception e) { e.printStackTrace(); }finally{ try{ ioreader.close(); }catch(Exception e){ e.printStackTrace(); } } return tmpNDCsvBean; } public static void writeSampleCSV (String outputFilePath, List listCSVBean){ Writer iowriter = null; CSVWriter writer = null; try { iowriter = new OutputStreamWriter(new FileOutputStream(outputFilePath),"UTF-8"); writer = new CSVWriter(iowriter, ',', CSVWriter.NO_QUOTE_CHARACTER); Iterator<samplecsvbean> ite = listCSVBean.iterator(); while(ite.hasNext()){ SampleCsvBean tt = ite.next(); String[] entries = new String[]{tt.getKey(), tt.getName(), tt.getPref(), tt.getJob().replace(",", "、") }; writer.writeNext(entries); } writer.close(); }catch (Exception e) { e.printStackTrace(); }finally{ try{ iowriter.close(); }catch(Exception e){ e.printStackTrace(); } } } } </samplecsvbean> |
一旦読み込むと、定義したJavaBeans型のリストとなるので、その後の加工は自由自在です。加工した結果を、csvに吐き出すコードは、上記の「writeSampleCSV」です。
一つ作っておくと、いろいろなcsvを加工するのも楽ですよ。
タグ：java
