python - 绘制遍历列名的 Pandas 数据框（作为变量）

Question

我正在尝试根据列名绘制具有不同设置的 Pandas 数据框。

我的数据框（topcount）是这样的

dateRep       Country1 Country2 Country3
20/03/2020        30       20      10
21/03/2020        40       28      15

我想在 y 轴上绘制列值，在 x 轴上绘制日期。

for country in list(topcount.columns):
    if country is 'Country3':
        topcount.plot(kind='line',x='dateRep',y='Country3',ax=ax,color='blue',linewidth=5)
    else:
        topcount.plot(kind='line',x='dateRep',y=country,ax=ax,linewidth=3)

问题在于它y=country不被解释为数据框列名。

我可以在这种情况下将列名作为变量传递还是正确的方法是什么？

标签： pythonpandasmatplotlib