python 리스트 컴프리핸션 잘 사용하기

for문 대신 많이 사용한다.

• 한줄만에 리스트 초기화,제작 가능!!

[] 리스트 생성

array = [i for i in range(5)]

# [0,1,2,3,4]

[] 조건 가진 리스트 생성

array = [i for i in range(5) if i % 2 ==0 ]

[] 특정원소 제외하고 리스트 생성

lst = [1,1,2,3,2,3,4,5,6]
remove_lst = {1,2,3}

array = [ i for i in lst if i not in remove_lst]
# [4,5,6]

[] 이차원 배열 생성

• 행 갯수 : 3 , 열 갯수: 5 인 2차원 배열 만들기

row = 3
column = 5
 
array = [[0]*column for _ in range(row)]
# [[0, 0, 0, 0, 0], 
# [0, 0, 0, 0, 0], 
# [0, 0, 0, 0, 0]]