Bedingte Daten in Python matplotlib hervorheben

Question

Ich habe einige Probleme, bedingt Pandas Datenrahmen plotten.

Sagen wir, wir haben einen 4 * 4 Datenrahmen, columns=['A', 'B', 'C', 'D'], indexes=[0, 1, 2, 3]. Nehmen wir an, wir machen eine df.plot.bar(). Und für die Daten in jeder Zeile möchte ich die Balken markieren, wenn Daten in column C >= column A + column B. Machen Sie dasselbe für Daten in Column D.

Das Endergebnis sollte die Balken, die die obige Anforderung erfüllen, in rot markieren, der Rest bleibt weiß.

Weiß jemand wie man es macht?

Answer 1

df = pd.DataFrame([ 
     [1, 2, 4, 2], 
     [4, 3, 5, 9], 
     [3, 1, 0, 1], 
     [2, 6, 9, 2] 
    ], columns=list('ABCD')) 

condAB = pd.DataFrame(False, df.index, ['A', 'B']) 
condCD = df[['C', 'D']].ge(df[['A', 'B']].sum(1), 0) 
cond = pd.concat([condAB, condCD], axis=1).stack().tolist() 
colors = np.where(cond, 'red', 'white') 

df.plot.bar(color=colors, legend=False)