2017-08-14 202 views
0

我在python中有一個函數,它使用for循環來生成一個數組;Python函數返回不正確的值

from pylab import * 

r0 = 3. 
radius = 3*r0 
signal_centre_i = randint(radius,100-radius) 
signal_centre_j = randint(radius,100-radius) 
bg_score = 30 
SN_ratio = 1. 
S0 = 10. 
signal_score = 0 
def generate_signal(): 
    signal_array = zeros((101,101)) 
    for i in range(101): 
     for j in range(101): 
      r = ((i-signal_centre_i)**2 + (j-signal_centre_j)**2)**0.5 
      signal_array[i,j] = signal_array[i,j] + S0*(1+((r/r0)**2))**-1.5 
    signal_score = 0 
    for i in range(101): 
     for j in range(101): 
      if ((i - signal_centre_i)**2 + (j-signal_centre_j)**2)**0.5 <= radius: 
       signal_score = signal_array[i,j] + signal_score 
      elif (((i - signal_centre_i)**2 + (j-signal_centre_j)**2)**0.5 <= radius +1) and ((i - signal_centre_i)**2 + (j-signal_centre_j)**2)**0.5 > radius: 
       signal_array[i,j] = 10 
    return signal_score 

generate_signal() 

while (((signal_score/bg_score)**0.5) < SN_ratio - SN_ratio/2) or (((signal_score/bg_score)**0.5) > SN_ratio + SN_ratio/2): 
    if (signal_score/bg_score)**0.5 > SN_ratio: 
     print "Calculated SN: "+str((signal_score/bg_score)**0.5) 
     S0 = S0 - S0/10 
     print "S0: "+str(S0) 
     print "Signal score: "+str(signal_score) 
     generate_signal() 
    elif (signal_score/bg_score)**0.5 < SN_ratio: 
     print "Calculated SN: "+str((signal_score/bg_score)**0.5) 
     S0 = S0 + S0/10 
     print "S0: "+str(S0) 
     print "Signal score: " +str(signal_score) 
     generate_signal() 

我真的很抱歉代碼的牆 - 如果有更好的方式來格式化,請讓我知道。

該代碼需要一個101x101陣列並在其上放置一個隨機「信號」(所以當你使用imshow()函數時,你會得到一個亮點)。但是,當我運行這段代碼時,signal score不會更新 - 我期望它在源中心周圍的一個圓內打印數組的總值,然後更新該方程,以使該值趨於預定值。但是,這不會發生,並且signal score停留在零。

我已經使用循環形式的

for i in range(101): 
    for j in range(101): 
     if ((i - centre_i)**2 + (j - centre_j)**2)**0.5 <= radius: 
      score = score + array[i,j] 

多次在過去出於同樣的目的,沒有任何問題。

我在做什麼錯?

+0

我已經糾正了錯誤的壓痕明顯 –

回答

2

您每次調用它時都忽略generate_signal函數的返回值。您需要將其分配給模塊範圍中的signal_score變量。

signal_score = generate_signal() 
+0

一旦你知道它...完全解決方案,我一直在尋找。謝謝! –