我使用scipy.weave.inline时遇到问题。我想编写一个以lcenter为中心的unittep函数,并使用width_nm。我有两个版本:python版本,名为pm和一个名为pm_weave的优化函数,但它看起来像abs不能正常工作。请参阅下面的代码。如果你运行它,你会得到一个大小为1的窗口,无论输入是什么,所以它看起来像abs不起作用。例如,如果删除abs,它的工作方式与预期完全相同
我该如何解决这个问题?
def pm_weave(w,lcenter,width_nm):
""" Return a unitstep function that is centered around lcenter with height 1.0 and width width_nm """
lcenter = float(lcenter)
w = float(w)
width_nm = float(width_nm)
code = """
#include <math.h>
float wl = 1.88495559215387594307758602997E3/w;
if(abs(lcenter-wl) < width_nm) {
return_val = 1.0;
}
else {
return_val = 0.0;
}
"""
res = weave.inline(code,['w','lcenter','width_nm'],
type_converters = weave.converters.blitz,
compiler = "gcc", headers=["<math.h>"]
)
return res
def pm(w,lcenter,width_nm):
"""
Return a unitstep function centered around lcenter [nm] with width width_nm. w
should be a radial frequency.
"""
return abs(600*np.pi/w - lcenter) < width_nm/2. and 1. or 0.
plot(wavelength_list,map(lambda w:pm(toRadialFrequency(w),778,1),wavelength_list),label="Desired behaviour")
plot(wavelength_list,map(lambda w:pm_weave(toRadialFrequency(w),778,1),wavelength_list),'^',label="weave.inline behaviour")
ylim(0,1.5)
show()
答案 0 :(得分:2)
我认为您可能需要在C代码中使用fabs()
而不是abs()
。 abs()
将截断结果,而fabs()
将用于浮点运算。