python - 让图例对应于 Pandas/PyPlot 中的 y 值

Question

我有一个名为“original_filtered_df”的数据框。我正在尝试为两列(“COLOR”、“SIZE”)和另外两列(“WEIGHT”、“HEIGHT”)随时间(“DATE”)绘制每对可能的值。

下面的代码绘制正确。它创建了多个图，每个图对应一个可能的值(“COLOR”、“SIZE”)。每个图包含两行，一行用于“WEIGHT”，另一行用于“HEIGHT”。然而，图例名称并没有说“WEIGHT”或“HEIGHT”。他们都说“DATE”(x 轴)。如何强制图例引用 y 轴数据，以便名称显示“WEIGHT”和“HEIGHT”？

import matplotlib.pyplot as plt

for combo in original_filtered_df.groupby(['COLOR', 'SIZE'], as_index=True):
    color = combo[0][0]
    size = combo[0][1]
    filtered_df = original_filtered_df[(original_filtered_df["COLOR"] == color) & (original_filtered_df["SIZE"] == size)]
    plt.figure()
    filtered_df.plot(x="DATE", y="WEIGHT", color=(1,0,0))
    filtered_df.plot(x="DATE", y="HEIGHT", color=(0,1,0))

    #patches, labels = ax.get_legend_handles_labels()
    #ax.legend(patches, labels, loc='center left', bbox_to_anchor=(1, 0.5))

    plt.legend(loc='center left', bbox_to_anchor=(1, 0.5))
    plt.title("COLOR " + str(color) + ", SIZE " + size)

Answer 1

我不确定为什么图例的标签默认为 x 轴，但我在绘图时通过执行以下操作解决了这个问题:

filtered_df.plot(x="DATE", y="WEIGHT", color=(1,0,0), label="WEIGHT")
filtered_df.plot(x="DATE", y="HEIGHT", color=(0,1,0), label="HEIGHT")

有了这个，图例函数使用了正确的 y 轴标签。