numpy初步使用

先来一段代码：

import numpy as np
import matplotlib.pyplot as plt

def f(t):
    '''cos和pi都是numpy中的函数，故需要用np.cos()和np.pi的形式
    '''
    return np.exp(-t)*np.cos(2*np.pi*t)
a=np.arange(0.0,5.0,0.02)   #第三个值表示取值间隔

plt.subplot(211)
plt.plot(a,f(a))

plt.subplot(2,1,2)
plt.plot(a,np.cos(2*np.pi*a),'r--')

plt.show()