Python 使用Pandas 创建的matplotlib图表中缺少标签

发布于11月22日

请考虑下面的例子

import pandas as pd
import matplotlib.pyplot as plt

ddd = pd.DataFrame({'time': [1,2,3,4],
                    'var1': [1,4,6,3],
                    'var2': [4,3,2,1]})


fig, ax = plt.subplots()

for col in ['var1', 'var2']:
    ddd.set_index('time')[col].plot(label = col)

这提供了:

但是，正如您所看到的，即使我在plot()调用中使用了label参数，图例的标签也不正确.你知道如何在保持循环 struct 的同时修复这个问题(在共享ax的同时按顺序添加图表)吗？

谢谢!

推荐答案

在代码中，您走在正确的道路上，但有一个小问题.您在循环中设置了索引，这就是导致标签问题的原因，因为DataFrame DDD的索引在每次迭代中都会更改.您可以在运行for循环之前对其进行设置，使其位于循环之外，如下所示，以使图例正确显示，并使两个列共享同一轴:

fig, ax = plt.subplots()

ddd.set_index('time', inplace=True)

for col in ['var1', 'var2']:
    ddd[col].plot(label=col, ax=ax)  # use the ax parameter to share the same axis

ax.legend()

plt.show()