python - 如何将数组转换为可调用函数

Question

我有一个 np.piecewise 函数，我想将其转换为可调用函数。

例如，假设我们有:

import numpy as np
x = np.linspace(0,10,1001)
my_func = np.piecewise(x, [x<8, x>=8], [np.sin, np.cos])

我有兴趣制作一个函数 my_callable_func，它对 my_func 有一些合理的评估。按理说，要么我们只是默认 x 中的前一步，要么我们在连续的 x 值之间使用某种线性近似值。

例如，在这种情况下 x = [0, 0.01, 0.02, ...]，所以给定 my_new_func(0.015)，我希望这样返回 np.sin(0.01) 或类似的东西...

Answer 1

您可以简单地将 np.piecewise 调用包装在一个函数定义中，

In [1]: def my_callable_func(x):
   ...:     return np.piecewise(x, [x<8, x>=8], [np.sin, np.cos])
   ...: my_callable_func(0.015)

Out[1]: array(0.01499943750632809)

原始 x 向量的值无关紧要。这将生成一个 0D numpy 数组，但如果需要，您可以将其转换为 float ，使用 return float(...)。