[Python] 리스트 컴프리헨션 List Comprehension

List Comprehension

리스트 안에 표현식(계산식)과 for문, if문을 한줄에 넣어서 새로운 리스트를 만드는 것입니다.

여기서 리스트는 list, tuple, dictionary, set 등 컨테이너 모두 해당합니다.

또한 표현식에 수식 뿐만 아니라 함수도 사용 가능합니다.

 

Example

리스트 생성

[ x for x in range(1, 10+1)]
#[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]

[ 2*x for x in range(1, 10+1)]
#[2, 4, 6, 8, 10, 12, 14, 16, 18, 20]

조건 걸기

[ x for x in range(1, 10+1) if x % 2 == 0]
#[2, 4, 6, 8, 10]

[ x for x in range(10) if x < 5 if x % 2 == 0]
#[0, 2, 4]

[ (x, z, y) for x in range(3) for y in ['x', 'y', 'z'] for z in ['가', '나']]
"""
[(0, '가', 'x'),
 (0, '나', 'x'),
 (0, '가', 'y'),
 (0, '나', 'y'),
 (0, '가', 'z'),
 (0, '나', 'z'),
 (1, '가', 'x'),
 (1, '나', 'x'),
 (1, '가', 'y'),
 (1, '나', 'y'),
 (1, '가', 'z'),
 (1, '나', 'z'),
 (2, '가', 'x'),
 (2, '나', 'x'),
 (2, '가', 'y'),
 (2, '나', 'y'),
 (2, '가', 'z'),
 (2, '나', 'z')]
"""