我在python中有一个函数,它使用for循环来生成数组;
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阵列并随机发出一个信号&#39;在它上面(所以当你使用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()