我有一个def分段(x,r,d,h,a,s)函数,该函数可创建2D形状。我需要围绕Y轴旋转它并创建一个3D地图,然后创建一个具有分段值的Z值和笛卡尔坐标中的X和Y值的值数组。
我的代码如下:
u = np.linspace(-self.extent / 2, self.extent / 2, (array_res + 1))
v = np.linspace(0, 2*np.pi, 60)
U, V = np.meshgrid(u, v)
defect = piecewise(U, self.radius, self.width, self.height, a, self.sigma)
x_pia = U * np.cos(V) * XRES / write_field + XRES / 2
y_pia = U * np.cos(V)* YRES / write_field + YRES / 2
x_mesh_pia, y_mesh_pia = np.meshgrid(x_pia, y_pia)
self.mill_map = np.transpose(np.array([np.round(defect),
np.round(x_mesh_pia),np.round(y_mesh_pia)])).astype(int)
其中XRES,YRES,write_field,array_res和所有self。属性在前面定义。我收到ValueError:设置具有序列错误的数组元素。我该如何解决?