javaで指定の行を取り出したい

Question

初めて質問するので、書き方が違いましたらすみません…

javaで指定の行を取り出したいです。

今できているところは
•csvファイルをJavaに読み込む
•csvファイルの中身は全て数値で、それらを配列に入れ込んだ
•結果をcsvファイルに出力する準備はできている

までで問題の中身なのですが、やりたいこととしては、
自分がコマンドで入力する数字と同じ行を持ってきてcsvファイルに入れ込みたいです。
（２　５　と入力したら２行目と５行目を取り出したいです。）

今書き終わっているところまでを貼ります。
教えて頂けないでしょうか？お願い致します。

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.File;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.FileReader;
import java.io.IOException;
import java.io.OutputStreamWriter;
import java.io.PrintWriter;
import java.util.Scanner;

public class aaa {

	public static void main(String[] args) {

		double[][] Data;
		double[][] kekkaData;
		String line;
		int a = 0;
		int b = 0;
		int N = 0;
		int h = 0;
		int tate = 2443;//csvの列数
		int yoko = 2334;//csvの行数
　　　　　　　   int hosii = 40;//取ってきたい行数の合計

		// データ格納
		try {
			File infile = new File("D:\Book3.csv");
			BufferedReader br = new BufferedReader(new FileReader(infile));

			while ((line = br.readLine()) != null) {
				N++;
			}
			br.close();
		} catch (FileNotFoundException e) {
			System.out.println(e);
		} catch (IOException e) {
			System.out.println(e);
		}
		Data = new double[tate][yoko];
		kekkaData = new double[hosii][yoko];


		try {
			File infile = new File("D:\Book3.csv");
			BufferedReader br2 = new BufferedReader(new FileReader(infile));
			File outfile = new File("D:\normdist_結果3.csv");
			PrintWriter bw = new PrintWriter(
					new BufferedWriter(new OutputStreamWriter(new FileOutputStream(outfile), "Shift-JIS")));
			while ((line = br2.readLine()) != null) {
				for (b = 0; b < yoko; b++) {
					String str[] = line.split(",", 0);
					Data[a][b] = Double.valueOf(str[b]);
					
				}
				a++;
			}
			br2.close();

			

//追加ここから
　　　　　　　Scanner scanner=new Scanner(System.in);
			int v = scanner.nextInt();
			System.out.println("抜き出したい行数は？"+v);
			
			for (int i = 1; i < tate+1; i++) {
				if( v == i){
				for (int j = 0; j < 1; j++) {
					System.out.println(kekkaData[i][j] + " , ");
					bw.write((kekkaData[i][j]) + ",");
				}
				System.out.println();
				bw.write("\r\n");
			
			}
			}

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

Comments

[jimbe]

コードのご提示には専用の書き方が用意されています. 入力枠の上辺にある <code> を押すと出てくる ``` で囲まれた「コード」という文字列を置き換える形で, コードを書き込み(コピペ)してください.
ご質問は編集できます. 入力枠の下にはプレビューも表示されますので, 確認しながら修正してみて頂けますか.

[uuu_]

教えて頂きありがとうございます。書き方を変更いたしました。
もしよろしければ、このＪａｖａについてのご回答を頂けると嬉しいです。お願い致します。

[jimbe]

> ２　５　と入力したら
入力部分がありません. これも作ってくれということでしょうか.

> int tate = 2443;//csvの列数
> int yoko = 2334;//csvの行数
行・列数を固定にしていますが, コード内には行数を数えている個所もあります. 行数は固定でしょうか可変でしょうか.

[jimbe]

すいません, コマンドで入力するということは args のことでしょうか.

[uuu_]

行数は固定です。
コマンドに入力するという表記で合っているのかが分からないのですが、「抜き出したい行数は？」と聞いて
２
５
と入力すると２、５行目が抜き出せるようにしたいです。お願いします。
今、追加でimport java.util.Scanner;と
Scanner scanner=new Scanner(System.in);
int v = scanner.nextInt();
System.out.println("抜き出したい行数は？"+v);を入れてみました。このｖと同じ行数を抜き出すようにしたいのですが、このままではうまくいかないのでご教授願います。お願い致します。

[jimbe]

指定は行数では無く行番号ということでよろしいですか.
追加されたコードでは 2 と 5 の二つを入力できませんし, 複数の数値を指定するのならば入力の終わりも指示しなくては, いつまでも指定し続けることになります. その辺りは如何でしょうか.

Answer 1

まず問題を考えてみましょう.
CSV として大量の数値が入っているファイルから, 指定した行を別ファイルに取り出します.
その為にご提示のコードでは CSV を解析して数値を一つ一つ取り出し, 配列に格納しています.

しかし必要なのは「行を取り出すこと」であり, "行"という構造があるテキストファイルでさえあれば, 内容が何であるかはプログラム的には考える必要はありません.
さらには, 行の指定が行番号であれば, 行を読む毎にそれが指定された行番号かを判断していけば, あらかじめ CSV 全体を読んでおく必要さえありません.

そこから考えると以下のようなコードが出来ます.

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

public class aaa {
  public static void main(String[] args) {
    String inFilename = "D:\Book3.csv";
    String outFilename = "D:\normdist_結果3.csv";

    List<String> nums = Arrays.asList(args);
    try(BufferedReader br = new BufferedReader(new FileReader(inFilename));
        PrintWriter pw = new PrintWriter(new FileWriter(outFilename));) {
      String line;
      for(int i=1; (line=br.readLine()) != null; i++) {
        if(nums.contains(""+i)) pw.println(line);
      }
    } catch(IOException e) {
      e.printStackTrace();
      System.exit(1);
    }
  }
}

---

追加
行を一つずつ指定する場合です.

import java.io.BufferedReader;
import java.io.FileReader;
import java.io.FileWriter;
import java.io.IOException;
import java.io.PrintWriter;
import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;

public class aaa {
  public static void main(String[] args) {
    String inFilename = "D:\Book3.csv";
    String outFilename = "D:\normdist_結果3.csv";

    //入力
    List<Integer> nums = new ArrayList<>();
    try(Scanner scanner = new Scanner(System.in);) {
      int v;
      while(true) {
        System.out.print("抜き出したい行は？(1以上を指定, 0以下で終了) :");
        v = scanner.nextInt();
        if(v <= 0) break;
        nums.add(v);
      }
    }
    if(nums.size() == 0) {
      System.out.print("抜き出しませんでした.");
      System.exit(0);
    }

    //抽出
    try(BufferedReader br = new BufferedReader(new FileReader(inFilename));
        PrintWriter pw = new PrintWriter(new FileWriter(outFilename));) {
      String line;
      for(int i=1; (line=br.readLine()) != null; i++) {
        if(nums.contains(i)) {
          System.out.println(line);
          pw.println(line);
        }
      }
      System.out.print("抜き出しました.");
    } catch(IOException e) {
      e.printStackTrace();
      System.exit(1);
    }
  }
}

Comments

[uuu_]

ありがとうございます（ ｉ _ ｉ ）
取り出すことができました。大変助かりました。ありがとうございました！