Python

Python_list comprehension

양쟁 2023. 3. 1. 15:38
a = [표현식 for 항목 in 반복 가능 객체 if 조건]

#'if 조건' 부분은 생략 가능
a = [표현식 for 항목1 in 반복 가능 객체1 if 조건1]
          for 항목2 in 반복 가능 객체2 if 조건2
          ...
          for 항목n in 반복 가능 객체n if 조건n]
          
          
#for문을 2개 이상 사용하는 것도 가능

 

예시

#짝수만 even list 변수에 저장

mylist = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]

even = [i for i in mylist if i % 2 == 0]
even
#홀수만 odd 리스트 변수에 저장

odd = [i for i in mylist if i % 2 == 1]
odd
# 1~100까지의 리스트를 생성하고, 
# 4의 배수만 result리스트에 저장하는 코드

result = [i for i in range(1, 101) if i% 4 == 0]
result
# 1~100까지의 리스트를 생성하고, 
# 4의 배수만 가져와서 2씩 곱하는 코드

result = [i*2 for i in range(1, 101) if i % 4 == 0]
resut
#list의 모든 값에 2를 더한 값을 변수 a에 저장

mylist = [1, 2, 3, 4, 5]

a = [i+2 for i in mylist]
a

'Python' 카테고리의 다른 글

Python_입출력  (0) 2023.03.01
Python_string method  (0) 2023.03.01
Python_반복문(for문/while문/range함수/format함수/continue/break)  (0) 2023.02.27
Python_조건문(If문)  (0) 2023.02.27
Python_비교/논리 연산자  (0) 2023.02.27