python array函数_python-如何将数组变成可调用函数

我有一个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)或类似的值…

解决方法:

您可以简单地将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,并返回float(…).

标签：python,numpy

来源： https://codeday.me/bug/20191120/2040947.html