由于广播错误,我无法使np.piecewise
用于多维绘图。在
有人有办法避开这个问题吗?在
以下是我在一个简化的可执行脚本中的内容:
import numpy as np
from pylab import *
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import axes3d, Axes3D
num_steps = 100
x_arr = np.linspace(0,100, num_steps)
y_arr = np.linspace(0,20, num_steps)
def zfunc(x, y):
return np.piecewise(x, [x>=500, x<500], [x, -x])
x,y = np.meshgrid(x_arr, y_arr)
z =zfunc(x,y)
fig=plt.figure()
ax=fig.subplot(1,1,1,projection='3d')
p = x.plot_surface(x,y,z,rstride=1,cstride=1,cmap=cm.coolwarm,linewidth=0,antialiased=False)
plt.show()
从而得出错误:
^{pr2}$
查看您使用的函数的docstring通常是一个好主意。我在那里找到了解决办法。在
相关问题 更多 >
编程相关推荐