NumPy数组比较报ValueError及np.piecewise使用报错求助

1. 第一个ValueError报错原因 #

你传入自定义函数f的x是长度为100的Numpy数组，执行x<0.5时会返回一个等长的布尔数组，而Python原生的if条件判断只能接收单个布尔值，无法直接解析数组的布尔值，因此触发歧义报错。

2. 第二个索引赋值报错原因 #

np.piecewise的第三个参数要求传入标量或者可调用对象（函数），你直接传入了预计算的全长度数组a*x和a*(1-x)（长度均为100），但每个条件掩码（比如x<0.5）仅匹配50个元素，无法将100个值赋值给50个目标位置，因此触发报错。

正确可运行代码 #

以下提供三种实现方案，均可正常输出倒V形折线图：

方案1：使用np.where实现（最简洁） #

import matplotlib.pyplot as plt
import numpy as np

x = np.linspace(0, 1, 100)
a = 2
y = np.where(x < 0.5, a * x, a * (1 - x))
plt.plot(x, y)
plt.show()

方案2：修正np.piecewise写法 #

将第三个参数替换为lambda函数，接收对应片段的x值做计算：

import matplotlib.pyplot as plt
import numpy as np

x = np.linspace(0, 1, 100)
a = 2
y = np.piecewise(x, [x < 0.5, x >= 0.5], [lambda t: a * t, lambda t: a * (1 - t)])
plt.plot(x, y)
plt.show()

方案3：向量化自定义函数 #

给原始自定义函数添加np.vectorize装饰器，使其支持数组输入：

import matplotlib.pyplot as plt
import numpy as np

x = np.linspace(0, 1, 100)
a = 2

@np.vectorize
def f(x):
    if x < 0.5:
        return a * x
    else:
        return a * (1 - x)

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

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