matplotlibの表面プロット（XY）^ 2

Question

は、誰もがf(x,y) = (x-y)^2

Z軸は関数f（x、y）を表すべき方程式の3次元表面プロットをプロットで私を助けることができる

私が持っています機能下記

def fnc(X): 
    return (X[0] - X[1]) ** 2 

ここでXは、X及びIは、特にそれがこのようである必要はY. としての第2として、第1のパラメータとnumpyのアレイです。ですから、署名を変更するように私に提案してはいけません。 ;）

私はthis溶液から以下試してみました：私は間違ったプロットを得るしかし

fig = plt.figure() 
ax = fig.add_subplot(111, projection='3d') 
x = y = np.linspace(-5,5,100) 
X, Y = np.meshgrid(x, y) 
Z = fnc1([np.linspace(-5,5,100) , np.linspace(-6,6,100)]) 
ax.plot_surface(X, Y, Z) 
ax.set_xlabel('X Label') 
ax.set_ylabel('Y Label') 
ax.set_zlabel('Z Label') 
plt.show() 

。

Answer 1

プロットする配列ZはXからの値の各ペアのためとYあなたが正確に一つのポイントを得るように、XとYが2Dアレイであると同じように、2次元アレイであるべきですZ。したがって、あなたのfnc関数への入力として、それらの配列XとYを使用することは理にかなって、Z = fnc([X,Y])

完全なコードが

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


def fnc(X): 
    return (X[0] - X[1]) ** 2 

fig = plt.figure() 
ax = fig.add_subplot(111, projection=Axes3D.name) 
x = y = np.linspace(-5,5,100) 
X, Y = np.meshgrid(x, y) 
Z = fnc([X,Y]) 
ax.plot_surface(X, Y, Z) 
ax.set_xlabel('X Label') 
ax.set_ylabel('Y Label') 
ax.set_zlabel('Z Label') 
ax.view_init(elev=15, azim=-118) 
plt.show() 

Answer 2

あなたfncは間違っています。表面はZ=(X-Y)**2とちょうど同じです。 Zのすべての計算がベクトル化されるため、最適解です。

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

fig = plt.figure() 
ax = fig.add_subplot(111, projection='3d') 
x = y = np.linspace(-5,5,100) 
X, Y = np.meshgrid(x, y) 
Z = (X-Y)**2 
ax.plot_surface(X, Y, Z) 
ax.set_xlabel('X Label') 
ax.set_ylabel('Y Label') 
ax.set_zlabel('Z Label') 
plt.show()