我有一个函数F
,定义为:
def F(x, A_0, A_1, phi_1, freq):
return (A_0 + A_1 * np.sin(2 * np.pi * 1 * freq * x + phi_1))
有时我希望将其扩展为这种形式:
def F(x, A_0, A_1, phi_1, A_2, phi_2, freq):
return (A_0 + A_1 * np.sin(2 * np.pi * 1 * freq * x + phi_1)
+ A_2 * np.sin(2 * np.pi * 2 * freq * x + phi_2))
我希望能够将该函数扩展到任意多个sin
项。从数学上讲,我希望函数返回:
执行此操作的 Python 方法是什么?
我的做法:
def F(x, A_0, As, phis, freq):
terms=[]
for i, A_i in enumerate(As):
terms.append(A_i*np.sin(2*np.pi*i*freq * x +phis[i]))
return A_0 + sum(terms)
如何做得更好?
最佳答案
您可以使用生成器表达式并避免创建中间列表:
def F(x, A_0, As, phis, freq):
terms = (A_i*np.sin(2*np.pi*i*freq * x +phis[i]) for i, A_i in enumerate(As))
return A_0 + sum(terms)
使用 NumPy 更好的想法是矢量化你的逻辑:
def F(x, A_0, As, phis, freq):
idx = np.arange(len(As))
terms = As * np.sin(2 * np.pi * idx * freq * x + phis)
return A_0 + terms.sum()
要了解其工作原理,请注意应用于数组的 np.sin
返回的数组的形状与将 np.sin
应用于每个 值。
正如 @Oleg 的评论中所述,您可能需要 np.arange(1, len(A_i)+1)
来确保您的枚举从 1
开始。
关于python - 从函数返回表达式之和的 Python 方式是什么?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/54514886/