Python函数返回不正确的值

时间:2017-08-14 14:16:36

标签: python arrays python-2.7 numpy

我在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]
过去很多次出于同样的目的,没有任何问题。

我做错了什么?

1 个答案:

答案 0 :(得分:2)

每次调用时,都会忽略generate_signal函数的返回值。您需要将其分配给模块范围中的signal_score变量。

signal_score = generate_signal()