matplotlib 기본: x, y의 할당

python에서 그래프를 그리기 위해서 matplotlib 모듈을 사용한다. 

이 모듈로 부터 그래프를 그리는 가장 기본은 x, y축을 설정하는 것이다. 다음을 보자. 

>>> import matplotlib.pyplot as plt

>>> x=range(100)

>>> y=[i**2 for i in x]

>>> plt.plot(x, y)

>>> plt.show()

다음은 sin 그래프로서 math 모듈을 사용한다. 

>>> import math 

>>> t=range(100)

>>> x=[(2*math.pi*i)/len(t) for i in t]

>>> y=[math.sin(i) for i in x]

>>> plt.plot(x, y)

>>> plt.show()

위와 같은 math.sin을 사용한 것과 같은 결과를 numpy 모듈을 사용하여 동일한 그래프를 그릴 수 있다. 

>>> import numpy as np

>>> x=np.linspace(0, 2*np.pi, 100)

>>> y=np.sin(x)

>>> plt.plot(x, y)

>>> plt.show()

다중 그래프를 하나의 플롯에 포함 

>>> x=np.array([0,1,2,4,5,6])

>>> y=np.array([0,1,4,16,25,36])

>>> z=np.array([6,5,4,3,2,1])

>>> data=np.c_[x,y,z]

>>> data

array([[ 0,  0,  6],

       [ 1,  1,  5],

       [ 2,  4,  4],

       [ 4, 16,  3],

       [ 5, 25,  2],

       [ 6, 36,  1]])

>>> data.shape

  (6, 3)

>>> for i in range(1, data.shape[1]):

>>> plt.plot(data[:,0], data[:,i])

>>> plt.show()