matplotlibによる3次元プロット

#背景
pythonで３次元プロットをする時、座標/関数をどのようなデータでプロットメソッドに渡せばいいか調べるのに時間がかかったため、備忘として残す。

#3次元プロットのコード
関数の定義

def function_2(x):
    return x[0]**2 + x[1]**2

3次元プロット

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

x0 = np.arange(-3, 3, 0.25)
x1 = np.arange(-3, 3, 0.25)
X, Y = np.meshgrid(x0, x1)
w = np.array([X,Y])
z = function_2(w)

fig = plt.figure()
ax = Axes3D(fig)
ax.set_xlabel("x0")
ax.set_ylabel("x1")
ax.set_zlabel("f(x0, x1)")
ax.plot_wireframe(X, Y, z)
plt.show()

#各配列の中身を理解する(ここが大事)
手書きですがご容赦を。