自作の関数でグラフを描く場合

変数を代入してグラフを描くのはカンタンだけれど、
意外と、関数にして同様のグラフを描く方法が思いつかなかった。

# 変数を代入する場合

import numpy as np
import matplotlib.pyplot as plt

x=np.arange(-10,11)
y=x**2

plt.plot(x,y)
plt.show()

#関数を使う場合
import numpy as np
import matplotlib.pyplot as plt

x=np.arange(-10,11)

def add(a):
return a**2

plt.plot(x,add(x))
plt.show()