[Python] List_Comprehension 리스트 안에서 for문, if문 사용하기

 

■ List_Comprehension

 

List_Comprehension은 조금 연습하면 반복문 등을 보다 편리하게 만들 수 있습니다.

 

 

1~10까지의 숫자 중에 5이상인 값만 추출하여 list로 저장하는 for문을 만들어 보겠습니다.

# 1~10까지 숫자 중에 5이상만 추출하는 for문
og_list=[1,2,3,4,5,6,7,8,9,10]

for i in og_list :
    if i >= 5  :
        print(i)


##### result #####   

5
6
7
8
9
10

 

# 1~10까지 숫자 중에 5이상만 list로 저장
og_list=[1,2,3,4,5,6,7,8,9,10]
new_list = []

for i in og_list :
    if i >= 5  :
        new_list.append(i)


##### result #####   

[5, 6, 7, 8, 9, 10]

↓↓ list관련 함수인 append 공부하기
https://planharry.tistory.com/14?category=855973

 

 

---

 

이제 위 예문들을 List_Comprehension을 사용하여 한 줄로 만들어 보겠습니다!

 

기본식은 아래와 같습니다.

[ 출력식 for 원소 in 리스트 if문 ]

# List_Comprehension을 사용하여 1~10까지 숫자 중에 5이상만 list로 저장
og_list=[1,2,3,4,5,6,7,8,9,10]

new_list = [ i for i in og_list if i >= 5]
new_list


##### result #####   

[5, 6, 7, 8, 9, 10]

 

출력되는 값을 식을 이용하여 변경 할 수도 있습니다.

# 출력식에 (i-4)를 넣어 1~6까지의 list를 만든다.
og_list=[1,2,3,4,5,6,7,8,9,10]

new_list = [ (i-4) for i in og_list if i >= 5]
new_list


##### result #####   

[1, 2, 3, 4, 5, 6]

위 예문을 변형하여 실습을 하다보면 충분히 이해하실 수 있을 겁니다.

 

 

 

---

 

lambda와 비슷한 점도 있기 때문에 함께 공부하시면 도움이 됩니다.

 

 

↓↓ lambda(람다) 내용 - 함수를 한 줄로 간단하게 표현하는 예약어

17. 파이썬 함수: 추가 형태와 기능(완벽 정리), lambda, global, args