파이썬 개념 - 리스트 내포(list comprehension)

list comprehension

파이썬(과 일부 언어)에서 리스트를 만들기 위해 사용 가능한 문법 구조 

리스트 재조합 시 유용

 

 

 

list comprehension 기본 형태

리스트이름 = [ 표현식 for 반복자 in 반복가능한것 ]

 

 

if문 포함한 list comprehension 형태

리스트이름 = [ 표현식 for 반복자 in 반복가능한것 if 조건문 ]

 

 

array = []
 
# for문 사용한 리스트 생성 방법
for i in range(0,20,2):
    array.append(i*i)
print(array)
 
# list comprehension으로 리스트 생성
array2 = [j*j for j in range(0,20,2)]
print(array2)
 
# list comprehension으로 조건부 리스트 생성
output = [k for k in array2 if k%5==0]
print(output)

실행 결과 ↓