[python] 파이썬 list comprehension - 리스트 컴프리헨션/조건부/예제

갑자기 이거 정리하다가

업무 연락와서 흐름이 끊겨버렸다 ☹️

 

🐌 기본 사용법 

# list comprehension - 읽기 쉽게 만들어줌
# 이전 리스트에서 새로운 리스트를 만들어줌
# new_list = [new_item for item in list]

nums = [1,2,3]
"""
new_list = []
for n in nums :
    new_n = n + 1
    new_list.append(new_n)
"""

# 위와 같은 코드
new_list = [n+1 for n in nums]

 

list comprehension 사용한 다양한 예제 

 

🐌 이름 알파벳 뽑기

# 이름 알파벳 리스트 
name = "Jimmy"
letters_list = [letter for letter in name]

print(letters_list)

🐌 실행 결과

 

 

🐌 1-5 2배수 리스트 만들기

# 1 부터 5까지 2배 되는 수 리스트 만들기
list = [n*2 for n in range(1,6)]

print(list)

🐌 실행 결과

 

🐌 조건부 list comprehension

# 조건부 list comprehension
# new_list = [new_item for item in list if test]

name_list = ["Jimmy", "Josh", "Max", "Teddy", "Jackson"]
# 4글자 이하 이름만 뽑기
new_list = [name for name in name_list if len(name)<=4]
print(new_list)

🐌 실행 결과