[파이썬][컴프리헨션] for문과 if문 한줄로 리스트 코딩하는 방법

안녕하세요~~

오늘은 파이썬 컴프리헨션 사용하는 방법에대해

포스팅하려고 합니다!

 

최근에 파이썬 공부를 시작했는데,

다른 사람들의 코드를 보며 약간 이해하기 쉽지 않은!

그런 코드가 있었는데 그것이 바로

파이썬 컴프리헨션이라는 문법이더라구요!

 

오늘 이 시간에 

그게 무엇인지 제대로 알아봅시다!!

 

파이썬 컴프리헨션을 사용하는 이유는?

다른 방법보다 속도도 빠르고,

직관적이기 때문에 

굉장히 많이 이용되고 있습니다!

 

for문을 한줄에 작성하기

temp = list([2, 3, 4, 5])
temp_new = list()
for i in temp:
    temp_new.append(i*2)

print(temp_new)

 

⬇️

 

temp = list([2, 3, 4, 5])
temp_new = list([i*2 for i in temp])

print(temp_new)

 

아래처럼 수정하게 되면,

결과는 같고 속도는 더 빠르게 됩니다.

 

temp는 동일하게 입력한 상태에서

for문을 한 줄로 작성하였습니다.

 

순서대로 작성한다고 생각하면 쉽습니다.

예를 들어, i는 2를 곱해서 리스트를 만드는데,

그 i는 for문으로 temp에서 가져와.

이런식으로요!!

 

if문을 추가하여 한줄에 작성하기

temp = list([2, 3, 4, 5])
temp_new = list()

for i in temp:
    if i>3:
        temp_new.append(i*2)
    else:
        temp_new.append(i)

위의 예제는 temp리스트에서 3보다 큰 경우에는 

2를 곱하고, 3보다 작은 경우에는 그냥

list에 넣는 코드입니다.

 

⬇️

 

list([i*2 if i>3 else i for i in temp])

 

다음과 같이 변경 가능합니다.

이 코드도 순서대로 읽는 것이 기억하기에 좋아요!

약간 영어 어순 읽듯이 해석하면 

좋을 듯 합니다.

 

---

 

어떤가요 

조금은 유용했나요~?

오늘은

파이썬의 컴프리핸션에 대해 알아보았습니다!

열공열공 합시다 💪