파이썬 딕셔너리 key 예외처리하기

파이썬 딕셔너리 key 예외처리하기

Python
파이썬 딕셔너리 key 예외처리하기
Author

gabriel yang

Published

October 2, 2024


파이썬에서 딕셔너리를 사용할 때, 특정 키(key)가 존재하지 않는 경우를 처리하는 방법은 매우 중요합니다. 일반적으로 딕셔너리에서 존재하지 않는 키를 조회하면 KeyError가 발생하지만, get() 메서드를 사용하면 이런 상황을 안전하게 처리할 수 있습니다.

이번 글에서는 파이썬의 get() 메서드를 활용해 딕셔너리에서 키가 없을 때의 상황을 효율적으로 처리하는 방법을 예시와 함께 설명하겠습니다.

1. 딕셔너리에서 키를 직접 조회할 때의 문제

파이썬 딕셔너리에서 특정 키에 접근하려면 대괄호([])를 사용합니다. 하지만 키가 존재하지 않으면 KeyError가 발생합니다.

예시: KeyError 발생

my_dict = {"name": "Alice", "age": 30}

# 존재하지 않는 키를 조회
print(my_dict["city"])  # KeyError: 'city'

출력

KeyError                                  Traceback (most recent call last)
Cell In[1], line 4
      1 my_dict = {"name": "Alice", "age": 30}
      3 # 존재하지 않는 키를 조회
----> 4 print(my_dict["city"])  # KeyError: 'city'

KeyError: 'city'
An error occurred while executing the following cell:
------------------
my_dict = {"name": "Alice", "age": 30}

# 존재하지 않는 키를 조회
print(my_dict["city"])  # KeyError: 'city'

위 코드에서는 "city"라는 키가 딕셔너리에 존재하지 않으므로 KeyError가 발생합니다. 프로그램이 이대로 종료되는 것을 방지하려면 오류 처리가 필요합니다.

2. get() 메서드를 사용한 안전한 조회

딕셔너리의 get() 메서드는 키가 존재하지 않을 때 기본값을 반환하여 KeyError를 방지할 수 있는 안전한 방법을 제공합니다.

get() 메서드의 기본 형식

my_dict = {"name": "Alice", "age": 30}
my_dict.get(key, default_value)
  • key: 조회할 키.
  • default_value (선택 사항): 키가 존재하지 않을 때 반환할 기본값. 이 값을 생략하면 None이 반환됩니다.

예시: get()을 이용한 안전한 조회

my_dict = {"name": "Alice", "age": 30}

# 키가 존재하면 해당 값을 반환
print(my_dict.get("name"))  # 출력: Alice

# 키가 없으면 None 반환 (기본값)
print(my_dict.get("city"))  # 출력: None

# 키가 없으면 기본값 반환
print(my_dict.get("city", "Unknown"))  # 출력: Unknown

출력

Alice
None
Unknown

위 코드에서는 "city"라는 키가 딕셔너리에 없지만, get() 메서드를 사용하면 KeyError 없이 안전하게 조회할 수 있고, 기본값을 설정할 수 있습니다.

3. get() 메서드를 활용한 실용적인 예시

get() 메서드는 실무에서 다양한 상황에서 유용하게 사용할 수 있습니다. 특히 딕셔너리에서 값이 선택적일 때나 기본값을 사용해야 할 때 매우 편리합니다.

예시 1: 사용자 설정값 적용

settings = {
    "theme": "dark",
    "language": "en"
}

# 기본값으로 안전하게 조회
theme = settings.get("theme", "light")
language = settings.get("language", "en")
timezone = settings.get("timezone", "UTC")

print(f"Theme: {theme}, Language: {language}, Timezone: {timezone}")

출력:

Theme: dark, Language: en, Timezone: UTC

이 예제에서는 사용자 설정값에서 "timezone"이 설정되지 않은 경우 get()을 사용하여 기본값인 "UTC"를 안전하게 적용할 수 있습니다.

예시 2: 카운터 구현

get() 메서드를 사용하면 딕셔너리를 활용한 카운터를 간단하게 구현할 수 있습니다. 만약 키가 딕셔너리에 없을 때 기본값을 0으로 설정해 증가시킬 수 있습니다.

words = ["apple", "banana", "apple", "orange", "banana", "apple"]

# 빈 딕셔너리 초기화
word_count = {}

for word in words:
    # 키가 없으면 0을 반환하고, 그 값을 1씩 증가시킴
    word_count[word] = word_count.get(word, 0) + 1

print(word_count)

출력:

{'apple': 3, 'banana': 2, 'orange': 1}

이 예시에서는 단어가 딕셔너리에 존재하지 않는 경우 get()을 통해 기본값 0을 반환하고, 카운팅 작업을 간편하게 처리합니다.

예시 3: 중첩된 데이터 Key 조회

딕셔너리에서 중첩된 데이터를 조회할 때도 get()을 사용하면 안전하게 접근할 수 있습니다.

data = {
    "user": {
        "name": "Alice",
        "profile": {
            "age": 30,
            "city": "New York"
        }
    }
}

# 중첩된 값에 안전하게 접근
age = data.get("user", {}).get("profile", {}).get("age", "Unknown")
city = data.get("user", {}).get("profile", {}).get("city", "Unknown")
country = data.get("user", {}).get("profile", {}).get("country", "Unknown")

print(f"Age: {age}, City: {city}, Country: {country}")

출력:

Age: 30, City: New York, Country: Unknown

이 예시에서는 딕셔너리 내부에 중첩된 값에 접근할 때 get()을 여러 번 사용하여 안전하게 값을 조회하고, 값이 없을 경우 기본값을 반환합니다.

4. get() 메서드를 사용할 때의 장점

  • 안전성: 존재하지 않는 키를 조회할 때 KeyError를 방지할 수 있어 프로그램이 중단되지 않고 계속 실행됩니다.
  • 가독성: 조건문을 사용하여 존재 여부를 확인할 필요 없이 기본값을 설정할 수 있으므로 코드가 간결하고 읽기 쉬워집니다.
  • 유연성: 기본값을 설정할 수 있어 다양한 상황에서 유연하게 사용할 수 있습니다.

5. get() 메서드 사용 시 주의사항

  • 성능: 딕셔너리가 매우 큰 경우 get() 메서드도 키를 찾기 위해 O(1) 시간 복잡도를 가지므로 성능에 큰 문제가 없습니다.
  • 잘못된 기본값 사용 주의: 기본값을 설정할 때 의도치 않은 기본값을 사용할 경우, 논리적 오류가 발생할 수 있으므로 주의해야 합니다. 기본값을 설정할 때는 실제 데이터와 잘 맞는 값을 설정하는 것이 중요합니다.

결론

파이썬의 get() 메서드는 딕셔너리에서 키가 없을 때 발생하는 문제를 효율적으로 처리할 수 있는 강력한 도구입니다. KeyError 없이 안전하게 데이터를 조회하고, 기본값을 설정하여 유연하게 사용할 수 있습니다. 또한 get() 메서드를 잘 활용하면 딕셔너리 사용 시 발생할 수 있는 오류를 방지하고, 코드의 가독성과 안전성을 높일 수 있습니다.