[프로그래머스] 짝수의 합

Oayenn·2023년 7월 29일
0

Programmers

목록 보기
6/6
post-thumbnail
post-custom-banner

잘 해결은 했지만 사용한 개념을 잊지 않기 위해 작성하는 포스트!

⚙️나의 코드

1.

def solution(n):
    sum = 0    
    for i in range(1, n+1):
        if i % 2 == 0:
            sum += i
    return sum

이 코드를 간결하게 만들 수 있는 방법은 없을까, 고민하고 찾아보다가 리스트 컴프리헨션(List Comprehension)을 알게 됐다. 바로 적용!

2.

def solution(n):
	return sum(i for i in range(1, n+1) if i % 2 == 0)

리스트 컴프리헨션sum()을 사용해서 함수를 간결하게 만들었다.
이참에 리스트 컴프리헨션에 대해서도 알아보자.

🔥List Comprehension

  • 파이썬에서 리스트를 생성하기 위한 간편한 기능. 반복문과 조건문을 조합해 한 줄로 리스트를 초기화할 수 있다.
  • 리스트 컴프리헨션의 일반적인 구조 :
    [표현식 for 항목 in 반복가능한_데이터 if 조건식]
  • 리스트뿐만 아니라 setdictionary 컴프리헨션도 지원한다.

일반적으로 for 루프를 사용해 리스트를 초기화하는 방법은 다음과 같다.

# 1부터 5까지의 제곱 값을 리스트에 저장하는 예제
squares = []
for i in range(1, 6):
    squares.append(i ** 2)
print(squares)  # Output: [1, 4, 9, 16, 25]

리스트 컴프리헨션을 사용하면 같은 코드를 한 줄로 간결하게 작성할 수 있다.

# 리스트 컴프리헨션을 이용한 예제
squares = [i ** 2 for i in range(1, 6)]
print(squares)  # Output: [1, 4, 9, 16, 25]
profile
차근차근 쌓아올리기
post-custom-banner

0개의 댓글