[java] OpenCSV를 사용하여 CSV 파일의 데이터를 정렬하는 방법은?

16 Nov 2023

java

OpenCSV는 Java에서 CSV 파일을 읽고 쓰는 데 사용되는 라이브러리입니다. CSV 파일의 데이터를 정렬하기 위해서는 몇 가지 단계를 따라야 합니다.

CSV 파일 읽기: 먼저 OpenCSV의 CSVReader 클래스를 사용하여 CSV 파일을 읽습니다. 다음과 같이 코드를 작성할 수 있습니다.

import com.opencsv.CSVReader;

try {
    CSVReader reader = new CSVReader(new FileReader("파일경로.csv"));
    String[] line;
    while ((line = reader.readNext()) != null) {
        // CSV 데이터 처리
    }
    reader.close();
} catch (IOException e) {
    e.printStackTrace();
}

데이터 정렬: CSV 파일의 데이터를 정렬하기 위해 Collection을 사용할 수 있습니다. 예를 들어, ArrayList를 사용하여 데이터를 저장하고 Collections.sort() 메서드를 사용하여 데이터를 정렬할 수 있습니다. 데이터를 정렬하기 전에 필요한 필드에 대한 Comparator를 구현해야 합니다.

import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
import java.util.List;

List<String[]> data = new ArrayList<>();

// CSV 데이터를 ArrayList에 추가
while ((line = reader.readNext()) != null) {
    data.add(line);
}

// 데이터를 정렬하기 위한 Comparator 구현
Comparator<String[]> comparator = new Comparator<String[]>() {
    @Override
    public int compare(String[] o1, String[] o2) {
        // 정렬 기준에 맞게 필드 비교 후 리턴
        // 예를 들어, 첫 번째 필드를 기준으로 오름차순 정렬하려면
        return o1[0].compareTo(o2[0]);
    }
};

// Comparator를 사용하여 데이터 정렬
Collections.sort(data, comparator);

정렬된 데이터 출력 또는 저장: 정렬된 데이터를 원하는 형식으로 출력하거나 다른 CSV 파일로 저장할 수 있습니다. 예를 들어, 정렬된 데이터를 콘솔에 출력하는 방법은 다음과 같습니다.

for (String[] row : data) {
    System.out.println(Arrays.toString(row));
}

또는 정렬된 데이터를 다른 CSV 파일로 저장하려면 OpenCSV의 CSVWriter 클래스를 사용할 수 있습니다.

import com.opencsv.CSVWriter;

try {
    CSVWriter writer = new CSVWriter(new FileWriter("출력파일경로.csv"));
    for (String[] row : data) {
        writer.writeNext(row);
    }
    writer.close();
} catch (IOException e) {
    e.printStackTrace();
}

위의 예제 코드를 통해 OpenCSV를 사용하여 CSV 파일의 데이터를 정렬하는 방법을 알아보았습니다. 필요에 따라 코드를 수정하여 원하는 결과를 얻을 수 있습니다.