python charts matplotlib 2, 2 dataframes

Question

I have some code which plots variables with the same names in 2 different dataframes;

    ax1.plot(df2[varname],'y-o',label='2')
    ax1.plot(df1[varname],'g-o',label='1')

in some cases df2 may not have the variable, in which case I just want to chart the variable in df1, so just one line, and not 2.

This however, produces an error if the variable isn't in both dataframes. Is there an obvious solution I'm missing?

Clément · Accepted Answer · 2019-11-06 10:47:00Z

1

Just test that varname exists in df2:

if(varname in df2.columns):
    ax1.plot(df2[varname],'y-o',label='2')
ax1.plot(df1[varname],'g-o',label='1')

answered Nov 6, 2019 at 10:47

Clément

1,2768 silver badges25 bronze badges

Sign up to request clarification or add additional context in comments.

1 Answer 1