如何在Python中计算分段函数？numpy数组逻辑判断报错求助

问题根源 #

你遇到的ValueError是因为numpy数组直接用普通if判断时，会生成布尔数组，而if无法判定整个布尔数组的真假，必须用数组级别的逻辑处理方式，或者改用普通列表逐个处理元素。

方案一：用numpy正确处理数组逻辑判断 #

利用numpy的布尔掩码或np.where实现矢量化分段计算，同时修正原代码中未导入三角函数/常量的问题：

方法1：布尔掩码（清晰直观） #

import numpy as np

# 生成0到1、步长0.02的数组
x = np.linspace(0, 1, num=51)
T_vc = 0.25
T_vr = 0.125

def vent_elastance(x):
    # 初始化结果数组，与输入同形状
    y = np.zeros_like(x)
    
    # 第一个区间：0 ≤ x ≤ T_vc
    mask_contraction = (x >= 0) & (x <= T_vc)
    y[mask_contraction] = 0.5 - 0.5 * np.cos(np.pi * x[mask_contraction] / T_vc)
    
    # 第二个区间：T_vc < x ≤ T_vc+T_vr
    mask_relaxation = (x > T_vc) & (x <= T_vc + T_vr)
    y[mask_relaxation] = 0.5 + 0.5 * np.cos(np.pi * (x[mask_relaxation] - T_vc) / T_vr)
    
    # 剩余区间保持0，无需额外处理
    return y

y = vent_elastance(x)

方法2：嵌套np.where（简洁紧凑） #

适合分段较少的场景：

import numpy as np

x = np.linspace(0, 1, num=51)
T_vc = 0.25
T_vr = 0.125

def vent_elastance(x):
    return np.where(
        (x >= 0) & (x <= T_vc),
        0.5 - 0.5 * np.cos(np.pi * x / T_vc),
        np.where(
            (x > T_vc) & (x <= T_vc + T_vr),
            0.5 + 0.5 * np.cos(np.pi * (x - T_vc) / T_vr),
            0
        )
    )

y = vent_elastance(x)

方案二：不用numpy，改用普通列表处理 #

如果不想依赖numpy，直接生成普通数值列表，逐个元素判断计算：

import math

# 生成0到1、步长0.02的列表（共51个元素）
x = [i * 0.02 for i in range(51)]
T_vc = 0.25
T_vr = 0.125

def vent_elastance(x_list):
    y = []
    for val in x_list:
        if 0 <= val <= T_vc:
            res = 0.5 - 0.5 * math.cos(math.pi * val / T_vc)
        elif T_vc < val <= T_vc + T_vr:
            res = 0.5 + 0.5 * math.cos(math.pi * (val - T_vc) / T_vr)
        else:
            res = 0
        y.append(res)
    return y

y = vent_elastance(x)

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