파이썬 list comprehension

Code Snippets 2021. 4. 28. 11:41

파이썬을 기초부터 공부하면서 느낀건데, 모듈을 이것저것 많이도 만들어 놓았다는 생각을 자주 한다.

이번에는 list comprehension이라는 녀석이다.

리스트 데이터를 만들 때 주로 사용하는 방법으로, for문 보다 빠르게 동작한다고 한다.

# 홀수만 추출하기
list = range(10)
odd_list = []
for num in list:
	if num % 2:
    	odd_list.append(num)

컴프리헨션을 사용하지 않는다면 홀수를 골라낼 때 위와 같이 써야하겠지만, 컴프리헨션을 쓰면 한 줄이면 끝난다.

# 컴프리헨션을 이용한 홀수 추출하기
odd_list = [num for num in range(10) if data % 2]

그치만 영 가독성은 떨어지는 듯?

admin