取两函数最大值的多元函数绘图报错求助

错误原因 #

你用的Python内置max()函数是用来找单个序列里的最大值，或者几个独立数值里的最大值，没法对两个numpy数组做逐元素的比较。当X和Y是meshgrid生成的二维数组时，np.cos(X)和np.sin(Y)都是二维数组，内置max()会尝试判断整个数组的"真假"，这就导致了报错里的"数组真值不明确"问题——你用if语句手动写max逻辑时，同样会遇到这个问题，因为if判断数组时也会触发同样的真值歧义。

解决方法 #

用numpy提供的np.maximum()函数，它专门做元素级别的最大值比较，能直接处理两个同形状的numpy数组，逐位返回对应位置的最大值。

修改后的代码如下：

import numpy as np
import plotly.graph_objects as go

def f(x,y):
    # 替换内置max为np.maximum，支持数组元素级比较
    return np.maximum(np.cos(x), np.sin(y))

x=np.linspace(0,5,20)
y=np.linspace(-3,2,20)
X, Y = np.meshgrid(x, y)
Z=f(X,Y)

fig = go.Figure(data=[go.Surface(x=X, y=Y, z=Z)])
fig.show()

补充说明 #

numpy的矢量化函数（比如np.maximum、np.minimum）都是为处理数组设计的，能避免循环和真值判断的问题，完全适配meshgrid生成的多维数组场景，这也是numpy处理数值计算的核心优势之一。

内容的提问来源于stack exchange，提问作者alice