绘制分段周期函数触发ValueError：如何解决数组真值判断歧义？

你写的f(x)是给单个数值设计的，但x是numpy生成的数组（包含1000个元素）。当用数组和标量做比较时（比如-np.pi < x < np.pi/2），会得到一个布尔数组，而if语句没法直接判断整个布尔数组的真假——它不知道该要求所有元素都满足（用.all()）还是至少一个满足（用.any()），所以抛出这个歧义错误。

以下是三种可行的修复方案：

方案1：用numpy向量化包装函数 #

把原函数转换成能处理数组的版本，自动遍历每个元素：

import matplotlib.pyplot as plt
import numpy as np

def f(x):
    if -np.pi < x < np.pi/2:
        return 0
    elif np.pi/2 < x < np.pi:
        return 1
    else:
        return np.nan  # 替换print，避免输出大量错误信息

# 包装成向量化函数
f_vec = np.vectorize(f)

x = np.linspace(-np.pi, np.pi, 1000)
plt.plot(x, f_vec(x))
plt.show()

注：vectorize本质是循环处理元素，适合简单场景，对性能要求高的话优先选其他方案。

方案2：用numpy专用分段函数piecewise #

这是最符合numpy数组操作逻辑的方法，效率更高：

import matplotlib.pyplot as plt
import numpy as np

x = np.linspace(-np.pi, np.pi, 1000)
# 定义分段条件和对应输出值
f = np.piecewise(x,
                 [(-np.pi < x) & (x < np.pi/2),
                  (np.pi/2 < x) & (x < np.pi)],
                 [0, 1])

plt.plot(x, f)
plt.show()

注意：numpy数组的逻辑与要用&，不能用and。

方案3：手动遍历数组元素 #

用列表推导式逐个处理x里的元素：

import matplotlib.pyplot as plt
import numpy as np

def f(x):
    if -np.pi < x < np.pi/2:
        return 0
    elif np.pi/2 < x < np.pi:
        return 1
    else:
        return np.nan

x = np.linspace(-np.pi, np.pi, 1000)
y = [f(i) for i in x]  # 逐个调用函数处理元素

plt.plot(x, y)
plt.show()

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