파이썬 리스트 컴프리헨션, 딕셔너리 컴프리헨션의 모든 것

 

 

리스트 컴프리헨션(List Comprehension)

파이썬 리스트 컴프리헨션은 간결하게 리스트를 생성하는 방법 중 하나로, 반복문과 조건문을 사용하여 리스트를 만들 수 있습니다. 리스트 컴프리헨션은 코드를 간결하게 유지하면서도 강력한 기능을 제공합니다.

일반적인 리스트 컴프리헨션의 구조는 다음과 같습니다:

[표현식 for 항목 in iterable if 조건]

- 표현식: 각 항목에 대한 계산식 또는 표현식을 나타냅니다.
- 항목: 반복 가능한(iterable) 객체에서 가져온 값입니다.
- iterable: 반복 가능한 객체(리스트, 튜플, 문자열 등)를 나타냅니다.
- 조건: 선택적으로 포함할 조건을 지정합니다. 조건은 필요에 따라 생략할 수 있습니다.

예를 들어, 0부터 9까지의 숫자 중에서 짝수만을 제곱하여 리스트를 만드는 경우:

squares = [x**2 for x in range(10) if x % 2 == 0]
print(squares)
# 출력: [0, 4, 16, 36, 64]

또 다른 예로, 문자열 리스트에서 문자열의 길이가 3 이상인 문자열만 선택하는 경우:

words = ["apple", "banana", "kiwi", "orange", "grape"]
filtered_words = [word for word in words if len(word) >= 3]
print(filtered_words)
# 출력: ['apple', 'banana', 'kiwi', 'orange', 'grape']

 

 

 

딕셔너리 컴프리헨션(Dictionary Comprehension)

파이썬 딕셔너리 컴프리헨션은 리스트 컴프리헨션과 유사하게 사용되며, 딕셔너리를 생성하는 간결한 방법을 제공합니다. 딕셔너리 컴프리헨션의 일반적인 구조는 다음과 같습니다:

{키_표현식: 값_표현식 for 항목 in iterable if 조건}

- 키_표현식: 각 항목에서 추출한 키를 나타내는 표현식입니다.
- 값_표현식: 각 항목에서 추출한 값을 나타내는 표현식입니다.
- 항목: 반복 가능한(iterable) 객체에서 가져온 값입니다.
- iterable: 반복 가능한 객체(리스트, 튜플, 문자열 등)를 나타냅니다.
- 조건: 선택적으로 포함할 조건을 지정합니다. 조건은 필요에 따라 생략할 수 있습니다.

간단한 예제를 통해 설명해보겠습니다. 1부터 5까지의 숫자를 키로 하고 각 숫자의 제곱을 값으로 하는 딕셔너리를 만드는 경우:

squares_dict = {x: x**2 for x in range(1, 6)}
print(squares_dict)
# 출력: {1: 1, 2: 4, 3: 9, 4: 16, 5: 25}

또 다른 예로, 문자열 리스트에서 각 문자열의 길이를 키로 하고 해당 문자열을 값으로 하는 딕셔너리를 생성하는 경우:

words = ["apple", "banana", "kiwi", "orange", "grape"]
length_dict = {word: len(word) for word in words}
print(length_dict)
# 출력: {'apple': 5, 'banana': 6, 'kiwi': 4, 'orange': 6, 'grape': 5}

 

딕셔너리에서 '전학생'을 키로 갖는 데이터만 제외하는 경우:

scores = {'철수': 50, '영희': 80, '길동': 90, '순희': 60, '전학생': 100}
scores = { name: score for name, score in scores.items() if name != '전학생'}
print(scores)

#출력: {'길동': 90, '순희': 60, '영희': 80, '철수': 50}

 

리스트, 딕셔너리 컴프리헨션은 코드를 간결하게 만들어주고, 여러 줄의 코드를 한 줄로 표현할 수 있어 가독성을 향상시켜줍니다.