더북(TheBook)

이번에는 배열에 연산이 아닌 함수를 적용하겠습니다.

import numpy as np
a = np.linspace(1, 2, 11)
print(np.sqrt(a))  # a값의 제곱근을 출력함

 

실행 결과

[1.         1.04880885 1.09544512 1.14017543 1.18321596 1.22474487
 1.26491106 1.30384048 1.34164079 1.37840488 1.41421356]

보다시피 배열에 어떤 연산이나 함수를 적용하면 배열의 모든 값이 한꺼번에 계산되는 것을 알 수 있습니다. 이런 특성을 활용하면 다음과 같이 아주 간단한 코드로 그래프를 나타낼 수 있습니다.

import matplotlib.pyplot as plt
import numpy as np
a = np.arange(-np.pi, np.pi, np.pi/100)
plt.plot(a, np.sin(a))
plt.show()

그림 13-8 numpy 배열에 sin( ) 함수를 적용해 그래프로 나타낸 결과

신간 소식 구독하기
뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.