"from pylab import *" は推奨されないとのことなので削除し"from numpy import linspace" を追加。その他それに合わせた変更。