使用定义的 function 绘制 pcolormesh 时出现 ValueError，其中包含 if 语句

Question

考虑以下代码：

import matplotlib.pyplot as plt
import seaborn as sns
import pandas as pd

x    = np.linspace(1,100,100)
y    = np.linspace(500,10000,100)

x    = np.unique(x)
y    = np.unique(y)
X,Y  = np.meshgrid(x,y)

Z    = myfunction(a,b,c,d,Y,e,f,g,X)

plt.pcolormesh(X,Y,Z)

plt.show()

其中a、b、c、d、e、f、g是我任意设定的值。 问题是，当我运行上述代码时，出现以下错误：

ValueError：具有多个元素的数组的真值不明确。 使用 a.any() 或 a.all()

因为我定义的 function 中有“if”语句。 我能做些什么来纠正这个问题？