調用函數時出現「UnboundLocalError：引用前引用的本地變量」

Question

對於繪圖，我基於條件（條件是熊貓數據框某些列中的某些值）定義顏色。現在我不確定，如果我在定義函數時犯了一個錯誤。 的功能如下：

def getColour(C, threshold): 
    neg = 0 - threshold 
    half = threshold/2 
    if C <= (neg - half): 
     clr = '#2b83ba' 
    if ((neg - half) < C <= neg): 
     clr = '#abdda4' 
    if ((threshold + half) > C >= threshold): 
     clr = '#fdae61' 
    if (C > (threshold + half)):  
     clr = '#d7191c' 
    return clr 

而且我這是怎麼實現它：我通過一個數據幀的行迭代，然後找到列滿足一個條件，使用指數從這些列從列表中獲取參數，應用另一個生成結果的函數（此函數起作用，當我爲繪圖設置固定顏色時腳本已經過測試並可正常工作），然後用不同的顏色繪製結果。

for index, row in Sparse.iterrows(): 
    lim = row[row.notnull()] 
    ci = [row.index.get_loc(x) for x in lim.index] 
    params = np.array(myList)[ci] 

    for i, w in enumerate(params): 
     w = w.tolist() 
     print w, w[2] 
     print ci[i] 
     colour = getColour(ci[i], threshold) 
     x, y = myFunction(w) 
     plt.plot(x,y, color=colour,linestyle='-',linewidth=1.5) 

但是，這將引發上線colour = getColour(ci[i], threshold)錯誤UnboundLocalError: local variable 'clr' referenced before assignment。

我已閱讀處理此錯誤的其他帖子，但我看不到我的問題是什麼。

Answer 1

一般來說，鞏固您的病情邏輯是一個好主意。 你的clr會有所有這些價值嗎？沒有，所以你可以使用一些elif：

def getColour(C, threshold): 
    neg = 0 - threshold 
    half = threshold/2 
    if C <= (neg - half): 
     clr = '#2b83ba' 
    elif ((neg - half) < C <= neg): 
     clr = '#abdda4' 
    elif ((threshold + half) > C >= threshold): 
     clr = '#fdae61' 
    elif (C > (threshold + half)):  
     clr = '#d7191c' 
    return clr 

然後做你的病情進行循環？你有沒有所有的情況下，如果？如果是，那麼在else中輸入錯誤是一個好主意。如果沒有，那麼這就是意味着你忘記了一個案例：

def getColour(C, threshold): 
    neg = 0 - threshold 
    half = threshold/2 
    if C <= (neg - half): 
     clr = '#2b83ba' 
    elif ((neg - half) < C <= neg): 
     clr = '#abdda4' 
    elif ((threshold + half) > C >= threshold): 
     clr = '#fdae61' 
    elif (C > (threshold + half)):  
     clr = '#d7191c' 
    else: 
     raise ValueError('Value expected for clr') 
    return clr 

編輯：爲了回答您的意見，我想你誤會我的意思。在Python中，如果出現意想不到的情況，最好拋出一個錯誤。 因此，要麼：

• 你給一個默認的顏色值，如「白」，你用它正常
• 將返回None，然後你的代碼的其餘部分應閱讀它之前檢查無 值（和那麼也許拋出錯誤）
• 您可以直接拋出一個錯誤

PEP20: Errors should never pass silently.