파이썬 라이브러리를 활용한 데이터 분석(2판): 영화 평점, 이름 통계, 선거 데이터 등 실사례 사용

163

4

장

NumPy 기본: 배열과 벡터 연산

그림

4-3

그리드에 적용한 함수의 그래프

4.3.1

배열 연산으로 조건절 표현하기

numpy

.

where

함수는

x

if

조건

else

y

같은 삼항식의 벡터화된 버전이다. 다음과 같은 불리

언 배열 하나와 값이 들어 있는 두 개의 배열이 있다고 하자.

In

[

165

]:

xarr

=

np

.

array

([

1

.

1

,

1

.

2

,

1

.

3

,

1

.

4

,

1

.

5

])

In

[

166

]:

yarr

=

np

.

array

([

2

.

1

,

2

.

2

,

2

.

3

,

2

.

4

,

2

.

5

])

In

[

167

]:

cond

=

np

.

array

([

True

,

False

,

True

,

True

,

False

])

164

파이썬 라이브러리를 활용한 데이터 분석(2판)

cond

의 값이

True

일 때는

xarr

의 값을 취하고 아니면

yarr

의 값을 취하고 싶다면 리스트 표기

법

list

comprehension

을 이용해서 다음처럼 작성할 수 있다.

In

[

168

]:

result

=

[(

x if c else y

)

.....:

for x

,

y

,

c in zip

(

xarr

,

yarr

,

cond

)]

In

[

169

]:

result ...