[java] 컬렉션 프레임워크를 활용한 데이터 분석

05 Dec 2023

java

컬렉션 프레임워크는 자바에서 데이터의 그룹을 다루기 위한 유용한 도구입니다. 이를 활용하여 데이터 분석 작업을 수행할 수 있습니다. 이번 포스트에서는 컬렉션 프레임워크를 사용하여 데이터를 저장, 조작 및 분석하는 방법에 대해 알아보겠습니다.

List

데이터 분석에서 가장 일반적으로 사용되는 컬렉션은 List입니다. List는 순서가 있는 데이터의 그룹을 저장하는 데 사용됩니다. List에는 중복된 값을 저장할 수도 있고, 순서가 유지됩니다.

다음은 List를 사용하여 데이터를 저장하는 예제입니다.

import java.util.ArrayList;
import java.util.List;

public class DataAnalysis {
    public static void main(String[] args) {
        List<String> data = new ArrayList<>();
        data.add("Apple");
        data.add("Orange");
        data.add("Banana");
        data.add("Apple");

        System.out.println(data);
    }
}

위의 예제에서는 ArrayList를 사용하여 데이터를 저장하고 출력합니다. 출력 결과는 다음과 같을 것입니다.

[Apple, Orange, Banana, Apple]

Set

Set은 List와 비슷하지만 중복된 값을 허용하지 않는 것이 특징입니다. Set은 데이터의 고유한 값들을 저장하고자 할 때 유용합니다.

다음은 Set을 사용하여 데이터의 고유한 값만 저장하는 예제입니다.

import java.util.HashSet;
import java.util.Set;

public class DataAnalysis {
    public static void main(String[] args) {
        Set<String> data = new HashSet<>();
        data.add("Apple");
        data.add("Orange");
        data.add("Banana");
        data.add("Apple");

        System.out.println(data);
    }
}

위의 예제에서는 HashSet을 사용하여 데이터를 저장하고 출력합니다. 출력 결과는 다음과 같을 것입니다.

[Apple, Orange, Banana]

Map

데이터 분석에서 특정 값을 키(key)와 함께 저장하고 싶을 때는 Map을 사용합니다. Map은 키와 값의 쌍으로 데이터를 저장하고, 특정 키를 사용하여 값을 검색할 수 있습니다.

다음은 Map을 사용하여 데이터를 저장하고 검색하는 예제입니다.

import java.util.HashMap;
import java.util.Map;

public class DataAnalysis {
    public static void main(String[] args) {
        Map<String, Integer> data = new HashMap<>();
        data.put("Apple", 10);
        data.put("Orange", 15);
        data.put("Banana", 5);

        int appleCount = data.get("Apple");
        System.out.println("Apple count: " + appleCount);
    }
}

위의 예제에서는 HashMap을 사용하여 데이터를 저장하고 특정 키를 사용하여 값을 검색합니다. 출력 결과는 다음과 같을 것입니다.

Apple count: 10

정리

이번 포스트에서는 컬렉션 프레임워크를 활용하여 데이터 분석 작업을 수행하는 방법에 대해 알아보았습니다. List, Set, Map 등의 컬렉션을 사용하여 데이터를 저장, 조작 및 검색할 수 있습니다. 데이터 분석 작업에 따라 적절한 컬렉션을 선택하여 활용하면 편리하고 효율적인 코드를 작성할 수 있습니다.

참고 문서:

Java Collections Framework - Oracle