диаграммы Python Matplotlib 2, 2 данных

Question

У меня есть некоторый код, который отображает переменные с одинаковыми именами в 2 разных фреймах данных;

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

в некоторых случаях df2 может не иметь переменной, и в этом случае я просто хочу отобразить переменную в df1, так что всего одна строка, а не 2.

Это, однако, приводит к ошибке, если переменная отсутствует в обоих кадрах данных. Есть ли очевидное решение, которое мне не хватает?

Answer 1

Просто проверьте, что varname существует в df2:

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