Python - Matplotlibで3次元グラフを書く

下記の式をグラフ化してみる。

[mathjax] $$f(x_0, x_1)=x_0²+x_1²$$

コード

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D

def func(x0, x1):
    return x0**2 + x1**2

x0 = np.arange(-3, 3, 0.25)
x1 = np.arange(-3, 3, 0.25)
X0, X1 = np.meshgrid(x0, x1)
Y = func(X0, X1)
fig = plt.figure()
ax = Axes3D(fig)
ax.plot_wireframe(X0,X1,Y)
plt.show()

参考：matplotlibで3Dグラフを描画する