やりたいこと

こんなCSVがあった場合、

game.csv

1,Splatoon 2,Nintendo,Nintendo Switch,2017年7月21日
2,Yakuza: Like a Dragon,SEGA,PlayStation 4,2020年1月16日
3,Undertale,Toby Fox,PlayStation 4,2015年9月15日
4,Fire Emblem: Three Houses,Nintendo,Nintendo Switch,2019年7月26日
5,Ghost of Tsushima,Sony,PlayStation 4,2020年7月17日

ある時は、↓こんな感じで2列目と3列目を取得したい。

Splatoon 2,Nintendo
Yakuza: Like a Dragon,SEGA
Undertale,Toby Fox
Fire Emblem: Three Houses,Nintendo
Ghost of Tsushima,Sony

でも場合によっては、↓こんな感じに1列目・2列目・5列目を取得したい。

1,Splatoon 2,2017年7月21日
2,Yakuza: Like a Dragon,2020年1月16日
3,Undertale,2015年9月15日
4,Fire Emblem: Three Houses,2019年7月26日
5,Ghost of Tsushima,2020年7月17日

つまり・・必要に応じて指定した列の値だけを抽出した～い！

つくった

つくりました。

GetSpecifiedColumn.java

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.File;
import java.io.FileReader;
import java.io.FileWriter;
import java.io.IOException;
import java.io.PrintWriter;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

/**
 * 指定したCSVから特定列の値のみを抽出します.
 * 抽出結果のファイルは「【抽出済】対象ファイル名」の名前で同階層に出力されます.
 *
 * 引数１：対象ファイル
 * 引数２：取得したい列番号（0始まり、カンマ区切りで複数指定が可能）
 */
public class GetSpecifiedColumn{

    public static void main(String[] args){

        // コマンドライン引数を取得.
        String path = null;
        List<String> colList = new ArrayList<String>();

        if (args.length == 2){
            path = args[0];
            colList = new ArrayList<String>(Arrays.asList(args[1].split(",")));
        } else {
            System.out.println("引数間違ってるよ！");
            return;
        }

        try{
            // 入力ファイル.
            File inFile = new File(path);
            BufferedReader br = new BufferedReader(new FileReader(inFile));

            // 出力ファイル.
            File outFile = new File("【抽出済】"+ path);
            PrintWriter pw = new PrintWriter(new BufferedWriter(new FileWriter(outFile)));

            // 列番号の最大値（最終列の番号）を取得.
            String lastCol = colList.get(colList.size() - 1);

            String line = null;
            // CSVファイルを1行毎に読み込む.
            while( (line = br.readLine()) != null) {
                String[] strList = line.split(",",0); // 行データを配列化.

                // 特定列だけを出力.
                for (String col : colList) {
                    if (!col.equals(lastCol)) {
                        // 最終列でない場合は、改行なし・カンマありで出力.
                        pw.print(strList[Integer.parseInt(col)]);
                        pw.print(",");
                    } else {
                        // 最終列の場合は、改行あり・カンマなしで出力.
                        pw.println(strList[Integer.parseInt(col)]);
                    }
                }
            }
            br.close();
            pw.close();

        } catch(IOException e) {
            System.out.println(e);
        }

        System.out.println("処理完了！YATTA！");
    }
}

うごかす

コンパイル

javac -encoding UTF-8 GetSpecifiedColumn.java

実行

java GetSpecifiedColumn game.csv 0,1,4

実行結果

【抽出済】game.csv

1,Splatoon 2,2017年7月21日
2,Yakuza: Like a Dragon,2020年1月16日
3,Undertale,2015年9月15日
4,Fire Emblem: Three Houses,2019年7月26日
5,Ghost of Tsushima,2020年7月17日

YATTA！

その後、無事業務データでも問題なく動作することを確認しました。わ～い！

補足

個人用途のため細々した制御は入れていません！ご注意ください。
※対象CSVのデータ部に「,（カンマ）」が含まれているとうまく動かない。存在しないCSVファイル名や、対象CSVに存在しない列番号を指定すると例外になるなど。

CSVファイルから特定列の値だけを抽出したい

やりたいこと

つくった

うごかす

補足