[python] 조건문에서의 세트 컴프리헨션 사용하기
들어가기
파이썬은 세트 컴프리헨션(set comprehension)이라는 기능을 제공하여 한 줄로 세트를 간편하게 생성할 수 있습니다. 세트 컴프리헨션을 사용하면 조건문으로 세트를 필터링할 수도 있어요. 이 기능을 통해 코드를 간결하게 작성할 수 있습니다.
세트 컴프리헨션
세트 컴프리헨션은 다음과 같이 구문을 작성합니다.
{표현식 for 항목 in 순회가능한객체}
위 구문에서 표현식은 각 항목에 대한 계산식이며, 순회가능한 객체는 리스트, 튜플, 문자열 등이 될 수 있습니다.
조건문과 함께 사용하기
세트 컴프리헨션을 사용하여 조건문을 추가할 수도 있습니다.
{표현식 for 항목 in 순회가능한객체 if 조건}
위와 같이 작성하면 조건이 참인 경우에 대해서만 세트를 생성합니다.
예를 들어, 1부터 10까지의 정수 중에서 짝수만 포함된 세트를 만들려면 다음과 같이 작성할 수 있습니다.
even_numbers = {x for x in range(1, 11) if x % 2 == 0}
print(even_numbers)
결론
세트 컴프리헨션을 사용하면 간결하고 가독성이 좋은 코드를 작성할 수 있습니다. 특히 조건문과 함께 사용하면 필요한 요소만을 간편하게 추출할 수 있습니다. 세트 컴프리헨션을 적절히 활용하여 파이썬 코드를 더욱 효율적으로 작성해보세요.
참고 문헌:
- https://realpython.com/list-comprehension-python/#set-comprehension