我有一个名为funcPower3的函数,其主体如下所示。我想用MatplotLib中的3D plot函数来绘制这个函数。我在使用meshgrid的scipy docs上看到了一个例子。但是,在本例中,函数不是定义的函数,而是一个简单的数学运算:fig = plt.figure()
ax = fig.gca(projection='3d')
X = np.arange(-5, 5, 0.25)
Y = np.arange(-5, 5, 0.25)
X, Y = np.meshgrid(X, Y)
R = np.sqrt(X**2 + Y**2)
Z = np.sin(R)
surf = ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap=cm.jet,
linewidth=0, antialiased=False)
我在这个示例中得到了灵感,但是我自己的函数签名与网格外的数据不兼容。我有这个错误:ValueError: zero-size array to minimum.reduce without identity
我的函数代码在这里:def funcPower3(PARAM):
inputX1 = open("power3X.txt","r")
inputY = open("power3Y.txt","r")
X1=[]
Y=[]
for line in inputX1:
X1.append(float(line))
for line in inputY:
Y.append(float(line))
resTmp_ = 0
res = 0
for i in range(len(X1)):
resTmp_ = Y[i] - (PARAM[0]*(X1[i])**float(PARAM[1]))
res += resTmp_**2
return res
以及绘制3d的代码此函数在这里:
xmin=0
ymin=0
x最大值=10
ymax=1fig = plt.figure('Power 3')
ax = fig.gca(projection='3d')
Z = []
Xpl = numpy.arange(xmin, xmax, 0.1).tolist()
Ypl = numpy.arange(ymin, ymax, 0.01).tolist()
Xpl, Ypl = numpy.meshgrid(Xpl, Ypl)
Z=[]
for i in range(len(Xpl[0])):
for j in range(len(Xpl)):
Z.append(funcPower3([Xpl[j][i],Ypl[j][i]]))
surf=ax.plot_surface(Xpl, Ypl, Z, rstride=8, cstride=8, alpha=0.3,cmap=cm.jet)
ax.set_xlabel('X Label')
ax.set_ylabel('Y Label')
plt.show()
感谢并问候您的建议;))