In [13]:
plt.plot(lin_space.reshape(500,1), cost_fcn.reshape(500,1), c = 'blue')
plt.scatter(w_array, cost_array, c='gray', s=15)
plt.scatter(w_array[-1], cost_array[-1], c = 'r', s =50)
plt.grid()
plt.xlabel('$\omega$')
plt.ylabel( r'$ J(\omega) $')
plt.title('Cost function')
plt.show()