[중급1] comprehension 연습

블로그에서 찾은 것¶

https://tariat.tistory.com/529?category=678799

for문 대체¶

In [2]:

temp = [2, 3, 4, 5]
temp

Out[2]:

[2, 3, 4, 5]

In [4]:

temp_new = list()
for i in temp:
    temp_new.append(i**2)
    
temp_new

Out[4]:

[4, 9, 16, 25]

In [6]:

temp_new_comp = [ i**2 for i in temp]
temp_new_comp

Out[6]:

[4, 9, 16, 25]

np.array()로 감싸면 어레이도 생성¶

In [9]:

import numpy as np

np.array( [ i**2 for i in temp] )

Out[9]:

array([ 4,  9, 16, 25])

for문 + if문 대체¶

• if else 는 for 문 앞에
  [ 연산 if 조건 else 연산2 for문 ]
• if만 있으 때는 for 문 뒤에
  [ 연산 for문 if 조건 ]

In [10]:

temp_new = list()

for i in temp:
    if i > 3:
        temp_new.append(i**2)
    else:
        temp_new.append(i)
        
temp_new

Out[10]:

[2, 3, 16, 25]

In [12]:

[ i**2 if i>3 \
else i \
for i in temp]

Out[12]:

[2, 3, 16, 25]

In [13]:

[ i**2 if i>3     else i      for i in temp]

Out[13]:

[2, 3, 16, 25]

In [15]:

[i**2 for i in temp if i>3]

Out[15]:

[16, 25]

In [ ]: