"UnboundLocalError: lokale Variable referenziert vor Zuweisung" beim Aufruf einer Funktion

Question

Zum Plotten definiere ich Farben basierend auf einer Bedingung (die Bedingungen sind einige Werte in bestimmten Spalten eines Pandas Datenrahmens). Jetzt bin ich mir nicht sicher, ob ich bei der Definition der Funktion einen Fehler mache. Die Funktion ist wie folgt:

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 

Und das ist, wie ich es umsetzen: Ich durch Zeilen eines Datenrahmens durchlaufen, dann finden Spalten eine Bedingung erfüllt, die Indizes aus diesen Spalten verwenden, um Parameter aus einer Liste, Wenden Sie eine andere Funktion an, die die Ergebnisse generiert (diese Funktion funktioniert, das Skript wurde getestet und funktionierte ordnungsgemäß, wenn ich eine feste Farbe für das Plotten festlegte) und dann die Ergebnisse mit einer anderen Farbe plotten.

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) 

Aber das wirft einen Fehler UnboundLocalError: local variable 'clr' referenced before assignment auf der Linie colour = getColour(ci[i], threshold).

Ich habe andere Beiträge zu diesem Fehler gelesen, aber ich kann nicht sehen, was mein Problem ist.

Answer 1

Im Allgemeinen ist es eine gute Idee, die Logik Ihrer Bedingung zu konsolidieren. Haben Ihre clr alle diese Werte? Nein, so können Sie einige elif verwenden:

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 

Dann haben Ihr Zustand geschlungen ist? Hast du alle Fälle in deinem if? Wenn ja, ist es eine gute Idee, einen Fehler in else zu werfen. Wenn nein, dann bedeutet das sind Sie ein Fall nur vergessen:

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 

Edit: Um auf deinen Kommentar zu beantworten, ich glaube, Sie mißverstehen, was ich meinte. In Python ist es besser, einen Fehler zu melden, wenn Sie etwas Unerwartetes haben. So entweder:

• Sie geben einen Standardfarbwert wie „White“ und Sie es verwenden normalerweise
• Sie kehren Keine und dann der Rest des Codes sollte keine Wert überprüfen, bevor es zu lesen (und dann vielleicht werfen Fehler)
• Sie direkt einen Fehler aus,

PEP20: Errors should never pass silently.