[Python3] 파이썬 리스트 선언 깔끔하고 빠르게 하기 (list comprehension)

배열 내 홀수 제거, 짝수 제곱 출력 문제

 

정수를 담은 리스트 mylist를 입력받아, 이 리스트의 원소 중 짝수인 값만을 제곱해 담은 새 리스트를 리턴하는 solution함수를 완성해주세요. 예를 들어, [3, 2, 6, 7]이 주어진 경우

 

• 3은 홀수이므로 무시합니다.
• 2는 짝수이므로 제곱합니다.
• 6은 짝수이므로 제곱합니다.
• 7은 홀수이므로 무시합니다.

따라서 2의 제곱과 6의 제곱을 담은 리스트인 [4, 36]을 리턴해야합니다.

 

위와 같은 문제를 해결하는 solution함수를 작성해보자.

 

mylist =  [3, 2, 6, 7]

def solution(mylist):
    answer = []

    for i in mylist:
      if i%2 == 0:
        answer.append(i**3) # pow(i, 3)
        
    return answer

print (solution(mylist))

 

 

위 소스코드는 직접 작성한 것이다.

 

아주 정상적인 코드라고 볼 수 있지만, 파이썬을 사용하여 구현한 의미를 살리기 위해 더욱 더 간단하고 깔끔하게 문제를 풀 수 있다.

 

list Comprehension - for 문과 if 문을 한 번에 처리

mylist = [3, 2, 6, 7]
answer = [ i**2 for i in mylist if i % 2 == 0]

 

아주 간단해진 코드를 볼 수 있다. 

 

list Comprehension을 이용하여 리스트의 선언 + 리스트 내 원소 대상 반복문 + 조건문을 동시에 한줄로 선언한다.

 

mylist의 원소 중 짝수인 경우에는 i를 제곱하여 answer 리스트에 삽입한다는 의미를 가진다.