💡|<반복 실행문>for<반복 변수>in<반복 범위>|
Q. 1~5까지 숫자가 들어있는 리스트에서 각 항목의 숫자를 제곱
numbers = [1, 2, 3, 4, 5]
square = []
for i in numbers:
square.append(i**)
print(square)
//[1, 4, 9, 16, 25]
numbers = [1, 2, 3, 4, 5]
square = [i**2 for i in numbers]
print(square)
//[1, 4, 9, 16, 25]
💡|<반복 실행문>for<반복 변수>in<반복 범위>|if<조건문>|
반복문을 수행하다가 조건문을 만족하는 경우에만 반복실행문을 실행
Q. 1~5까지 숫자가 들어있는 리스트에서 3이상의 숫자만 제곱
numbers = [1, 2, 3, 4, 5]
square = []
for i in numbers:
if i >= 3:
square.append(i**)
print(square)
//[9, 16, 25]
numbers = [1, 2, 3, 4, 5]
square = [i**2 for i in numbers if i >= 3]
print(square)
//[9, 16, 25]
컴프리헨션 방법을 이용하면 훨씬 간결하게 작성할 수 있음 ㅎㅎ
출처 : 데이터분석을 위한 파이썬 철저 입문(최은석)