20201110 DAY08 3차원 그래프 그리기

숫자 8을 2차원 행렬의 색상으로 표현하기(plt.pcolor 함수)

(출처: 《파이썬으로 배우는 머신러닝의 교과서》, 코드는 멘토님이 제공)

 

원래 코드 

import numpy as np
import matplotlib.pyplot as plt
 
#함수 f3을 정의
def f3(x0, x1): 
    r = 2 * x0**2 + x1**2
    ans = r * np.exp(-r)
    return ans
 
#x0, x1에서 각각 f3을 계산
xn = 9
x0 = np.linspace(-2, 2, xn)
x1 = np.linspace(-2, 2, xn)
 
y = np.zeros((len(x0), len(x1)))
#9*9 좌표 생성
for i0 in range(xn):
    for i1 in range(xn):
        y[i1, i0] = f3(x0[i0], x1[i1])
 
print(x0)
print(np.round(y, 1))
plt.figure(figsize = (3.5, 3))
plt.gray()
plt.pcolor(y)
plt.colorbar()
plt.show()

 

이래저래 갖고 놀아본 결과