python如何使float-to-float函数在列表上按元素运行？

Question

我有以下代码：

def tau_r(u):
    return (u_1 < u < u_2) * (T1 - T2) + T2
if __name__ == "__main__":
    figure()
    plot(u, tau_r(u))
    show()

运行时出现以下错误：

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

我想这是因为它将u视为一个列表，而它应该分别对列表中的每个元素执行任务。 我怎样才能解决这个问题？你知道吗

Answer 1

通过使用map，可以对列表的所有元素应用相同的函数。它将返回一个包含函数返回值的新列表。你知道吗

plot(u, list(map(tau_r, u)))