前回に引き続き、3次元のグラフの表示をしました。
前回までは、Jupyter Notebookでプログラミングしましたが、今回はVisual Studio Code (以下、VSCode)で実行しました(補完やコード整形があって便利)。
2変数関数①初めに作成した関数
以下、プログラム
from mpl_toolkits.mplot3d import Axes3Dimport numpy as npimport matplotlib.pyplot as pltimpor
数値を色で表現する初めに、「【Python】等高線の表示」でも使った、以下の2変数関数を定義する。
import numpy as npimport matplotlib.pyplot as plt%matplotlib inlinedef f3(x0,x1): ans = (2 * x0**2 + x1**2) * np.exp(-(2 * x0**2 + x1**2)) return ans
9x9の0で初期化された行列yを用意して、各要素に関数f3()の演
【Python】 等高線の表示
関数contour()を使うxn=50x0=np.linspace(-2,2,xn)x1=np.linspace(-2,2,xn)y=np.zeros((xn,xn))for i0 in range(xn): for i1 in range(xn): y[i1,i0]=f3(x0[i0],x1[i1])xx0,xx1=np.meshgrid(x0,x1)plt.figure(1,figsize=(4,4))cont=plt.contour(xx0,xx1,y,