리스트 내포(List comprehension)

[표현식 for 항목 in 반복가능객체 if 조건문]

1. 기본 예제

list_A 리스트의 각 항목에 3을 곱한 결과를 result 리스트에 담아보자.

list_A = [1, 2, 3, 4]

(1) 리스트 안에서 for문 사용하기

result = []
for num in list_A:
  result.append(num*3)
print(result)

# output
[3, 6, 9, 12]

(2) 리스트 내포 사용하기

result = [num*3 for num in list_A]
print(result)
# output
[3, 6, 9, 12]

2. 응용

(1) 리스트 내포 안에 if 조건 사용하기

[표현식 for 항목 in 반복가능객체 if 조건문]

result = [num*3 for num in list_A if num > 2]
print(result)
# output
[9, 12]

(2) for문 여러개

[표현식 for 항목1 in 반복가능객체1 if 조건문1  
        for 항목2 in 반복가능객체2 if 조건문2  
        ...  
        for 항목n in 반복가능객체n if 조건문n]

# 구구단 만들기 (1단, 2단)
result = [x*y for x in range(1, 3)
              for y in range(1,10)]
print(result)
# output
[1, 2, 3, 4, 5, 6, 7, 8, 9, 2, 4, 6, 8, 10, 12, 14, 16, 18]

출처 점프투파이썬