ValueError: truth value of array is ambiguous 数值错误：数组的真值是模糊的

10 浏览
0 Comments

ValueError: truth value of array is ambiguous 数值错误：数组的真值是模糊的

代码：

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
# 定义网格
r = np.linspace(0.02, 0.98, 10)
theta = np.linspace(0, 2*np.pi)
rgrid, thetagrid = np.meshgrid(r, theta)
xgrid, ygrid = rgrid*np.cos(thetagrid), rgrid*np.sin(thetagrid)
# 定义计算z值的函数
def get_zvalue(r):
    if np.any(r<0.03):
        return np.zeros_like(r)
    elif np.any(r>0.02):
        delta_s = 0.02/np.sqrt(1-(r-0.02)**2)
        return delta_s + get_zvalue(r-0.02)
# 计算zsurface
zsurface = get_zvalue(r)
# 绘制图像并保存
fig = plt.figure()
ax = fig.gca(projection='3d')
surf = ax.plot_surface(xgrid, ygrid, zsurface)
plt.savefig("Hoops.png")

错误：

ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()

这个 ValueError 是由于函数 get_zvalue 中的 "if r<0.03:" 这一行触发的。我对 numpy 不太熟悉，不知道如何处理这个错误。有没有其他方法可以创建 zsurface 数组而不会出现这个错误？

0