[Python/파이썬] 리스트, while 문을 이용하여 총합계 구하기 / list comprehension

* 본 포스팅은 주피터 노트북을 이용하였다. 이 코드로는 파이참에서도 실행이 가능하다.

 

1. while 문을 이용하여 총합계 구하기

arr = [1,2,3,3,7,9,23]
sum=0
i=0
while i < len(arr):
    sum = sum +arr[i]
    i += 1
print(sum)

 

2. list comprehension(리스트 내포)

[출력표현식 for 요소 in 입력Sequence [if 조건식]]

예제를 풀어보자

arr = [1,2,3,3,7,9,23,100,4,12,7]
arr1=[i*i for i in arr] #list comprehension
print(arr1)

---

이번에는 기존의 list 예제와 list comprehension을 비교해보자

문제는 arr 리스트 안에 짝수만 새로운 리스트(arr1)에 넣는 것이다.

 

기존 list

arr = [1,2,3,3,7,9,23,100,4,12,7]
arr1=[]
sum=0
i=0
while i < len(arr):
    if arr[i]%2==0:
        arr1.append(arr[i])
    i += 1
print(arr1) #[2, 100, 4, 12]

list comprehension

arr = [1,2,3,3,7,9,23,100,4,12,7]
arr2 = [i for i in arr if i%2==0]
print(arr2)