将IDL代码翻译为Python

Question

我正在尝试使用Python编写的IDL代码通过其标准偏差函数中的一系列值计算背景噪音。下面的代码：

; (INPUT) 
; data = 1-D array of intensities 
; (OUTPUT) 
; bck,sig = background and fluctuations (1 sig level) 
; ndata = number of values upon which bck,sig are computed 
; POSITIVE = retains only data values > 0 
; NSIG = number of sigmas 

    if not(keyword_set(NSIG)) then nsig=3.  
    if keyword_set(POSITIVE) then begin 
    test=where(data gt 0.) 
    if test(0) ne -1 then data2=data(test) else begin 
     bck=0 & sig=-1 & ndata=0 & goto,out 
    endelse 
    endif else data2=data 
    if n_elements(data2) gt 1 then begin 
    bck=mean(data2,/double) & sig=stddev(data2,/double) 
    endif else begin 
    bck=data2(0) & sig=0 & ndata=1 & goto,out 
    endelse 
    ndata=1 

loop: 
    test=where(abs(data2-bck) lt nsig*sig) 
    if n_elements(test) eq 1 then goto,out 
    ndata=n_elements(test) 
    moy=mean(data2(test),/double) 
    sig=stddev(data2(test),/double) 
    if moy eq bck then goto,out 
    bck=moy 
    goto,loop 

out: 
    return 
end 

代码的心脏是循环的，这里是我试图复制它：

def bg(array): 
    temp = [] 
    for i in range(len(array)): 
     if array[i]-np.mean(array) < 3*np.std(array): 
      temp.append(array[i]) 
    avg=mean(temp) 
    return avg 

这是正确的吗？原始代码是否只能找到那些低于3 * std的值的平均值？

这一行中的1实际上意味着什么？ if n_elements(test) eq 1 then goto,out

Answer 1

看起来这就是代码正在做的事情。它只会计算在nsig之下的点的平均值。

if n_elements(test) eq 1 then goto,out表示如果测试中只有一个元素，则转至out行，然后执行返回。

一般而言，goto语句应该在IDL中避免，它们有使代码非常混乱且难以调试的倾向。