python 分段函数 表示曲面_绘制三维分段函数

我是python的新手，正在尝试3d绘制分段函数。我试图在z轴上绘制下面的“主公式”函数，因为它随x和y的变化而变化，范围从0到10，常数=1。但我似乎不太明白这里的作图方法。在from sympy import *

from mpl_toolkits.mplot3d import Axes3D

from matplotlib import cm

from matplotlib.ticker import LinearLocator, FormatStrFormatter

import matplotlib.pyplot as plt

import numpy as np

def mainformula(x,y,constant):

return Piecewise((subformula1(x,y,constant), y >= 0 and y < 3),(subformula2(x,y,constant),y>=3 and y <= 10))

def subformula1(x,y,constant):

return x + y + constant

def subformula2(x,y,constant):

return x - y - constant

fig = plt.figure()

ax = fig.gca(projection='3d')

X = np.arange(0, 10, 0.25)

Y = np.arange(0, 10, 0.25)

constant = 1

X, Y = np.meshgrid(X, Y)

Z = mainformula(X,Y,constant)

surf = ax.plot_surface(X, Y, Z)

plt.show()

运行该代码时，我得到的错误是：“ValueError：具有多个元素的数组的真值不明确。使用a.any()或a.all()“