컴프리헨션 (Comprehension) - 리스트 내 for문, if문 사용하기

 

list1 = list()
for i in range(5):
    list1.append(i)
print(list1)

[0, 1, 2, 3, 4]

리스트를 구현할 때 위처럼 리스트를 선언한 후 for문을 따로 돌려서 리스트를 초기화할 수도 있지만

리스트 내에서 for문을 선언하여 아래와 같이 한 줄로 초기화 할 수도 있음 = 리스트 컴프리헨션 (List Comprehension)

list1 = [i for i in range(5)]
print(list1)

[0, 1, 2, 3, 4]

맨 위에 있는 코드 세 줄을 한 줄로 줄일 수 있다.

 

alpha = [chr(i) for i in range(97, 123)]
loc = [-1 for i in range(len(alpha))]
print(loc)

[-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1]

위처럼 숫자를 넣어서 해당 숫자로 전체를 초기화 할 수도 있음

 

even = [i for i in range(1, 11) if i % 2 == 0]
print(even)

[2, 4, 6, 8, 10]

리스트 안에 if문을 넣어서 조건을 걸 수도 있다.

 

 

for문과 if문은 얼마든지 중복해서 사용 가능하며 집합과 딕셔너리에서도 똑같이 적용된다.