Python绘制分段函数遇数组真值判断模糊错误，求解决方案

错误原因分析 #

1. 自定义函数的问题：你写的S(x)是针对单个数值设计的，但传入的是numpy数组。当用0 <= x <=1判断数组时，会生成一个布尔数组，Python无法直接判断整个数组的“真值”，因此抛出The truth value of an array with more than one element is ambiguous错误。
2. np.piecewise的问题：你的条件存在重叠（比如x<=1和1<=x<=2都包含x=1，1<=x<=2和x>=2都包含x=2），np.piecewise要求条件互斥，重叠会导致逻辑歧义。

修正方案 #

方案一：修改自定义函数，支持数组输入

通过数组掩码（mask）为不同区间的元素赋值，避免直接用if判断整个数组：

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(0, 3, 1000)

def S(x):
    # 初始化和x同形状的结果数组
    res = np.zeros_like(x)
    
    # 处理0<=x<=1区间
    mask = (0 <= x) & (x <= 1)
    res[mask] = -1 + x[mask] - (5/9)*x[mask]**2 + (5/9)*x[mask]**3
    
    # 处理1<x<=2区间
    mask = (1 < x) & (x <= 2)
    res[mask] = (14/9)*(x[mask]-1) + (10/9)*(x[mask]-1)**2 - (2/3)*(x[mask]-1)**3
    
    # 处理2<x<=3区间
    mask = (2 < x) & (x <= 3)
    res[mask] = 2 + (16/9)*(x[mask]-2) - (8/9)*(x[mask]-2)**2 - (1/9)*(x[mask]-2)**3
    
    # 边界点x=1和x=2，两个区间的表达式值一致，直接赋值
    res[x==1] = 0
    res[x==2] = 2
    return res

plt.plot(x, S(x), 'k')
plt.show()

方案二：修正np.piecewise的条件，确保互斥

调整条件为完全不重叠的区间，让np.piecewise能正确匹配每个元素：

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(0, 3, 1000)

y = np.piecewise(x,
                 # 互斥的区间条件
                 [x <= 1, (x > 1) & (x <= 2), x > 2],
                 # 对应区间的表达式
                 [lambda x: -1 + x - (5/9)*x**2 + (5/9)*x**3,
                  lambda x: (14/9)*(x-1) + (10/9)*(x-1)**2 - (2/3)*(x-1)**3,
                  lambda x: 2 + (16/9)*(x-2) - (8/9)*(x-2)**2 - (1/9)*(x-2)**3])

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

验证边界点 #

你可以自行验证x=1和x=2处的函数值：

• x=1时，第一个表达式结果为-1 +1 -5/9 +5/9=0，第二个表达式结果为0+0-0=0，值一致；
• x=2时，第二个表达式结果为14/9 +10/9 -6/9=18/9=2，第三个表达式结果为2+0-0-0=2，值一致。因此边界点无需额外处理也能保证连续。

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