본문 바로가기
카테고리 없음

[파이썬을 파이썬답게] for 문과 if 문을 한번에

by YGSEO 2021. 3. 8.
728x90

2-depth 블록

def solution(mylist):
    answer = []
    for a in mylist:
        if a % 2 == 0:
            answer.append(a**2)
        else:
            continue
    return answer

list comprehension

한줄에

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

댓글