[Python] 리스트 컴프리헨션 list comprehension

기존 배열

아래는 어떤 리스트에 3을 곱한 뒤 새로운 리스트를 만드는 코드이다.

a=[1,2,3,4]
result=[]
for num in a:
  result.append(num*3)

 

리스트 컴프리헨션 적용 배열

리스트 컴프리헨션을 사용하면 아래와 같이 코드 길이를 한 줄로 줄일 수 있다.

a=[1,2,3,4]
result=[num*3 for num in a]
print(result)

 

조건에 따른 리스트 컴프리헨션

if로 필터링을 걸 수도 있다.

result=[num*3 for num in a if num%2==0]

 

for 문 여러개 사용하기

for문을 두 개 사용하여 배열을 생성할 수도 있다. 하지만 두 개 초과하여 사용하지 말자.

#for문 여러개 사용하는 경우
result =[x*y for x in range(2,10)
for y in range(1,10)]
print(result)

리스트 컴프리헨션의 기본 문법

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

 

배운 점

분명 알고 있는 문법인데 실제로 코드를 작성할 때 잘 활용하지 못하는 것 같다. 현재 작성한 코드들 다시 보면서 이렇게 고칠 수 있으면 고쳐야겠다.