基于多索引级别为绘图的不同部分着色

Question

我需要使用 pyplot 创建一个绘图，在其中我可以根据 DataFrame (df) 中的第一个索引级别为标记着色。 示例：以下所有值：A -- 红色、B -- 绿色、C -- 蓝色 我正在针对 cat2 值绘制 a1、a2、a3、a4、a5。每个值都应该在同一个图上。

我可以通过使用横截面使用子图来做到这一点：

f, a = plt.subplots(3,1)
df.xs('A').plot(ax=a[0])
df.xs('B').plot(ax=a[1])
df.xs('C').plot(ax=a[2])

子图可以让我为不同的 xs 着色，但我需要一种方法来在同一个图上进行。 Link is an image of my dataframe in csv

Answer 1

我想这样的事情会奏效。

f, a = plt.subplots(1, 1)
# Only create one subplot as all the data is meant to go on one plot only
df.xs('A').plot(ax=a, marker='D', markersize=6, markerfacecolor='red',
                markeredgecolor='red')
df.xs('B').plot(ax=a, marker='D', markersize=6, markerfacecolor='green',
                markeredgecolor='green')
df.xs('C').plot(ax=a, marker='D', markersize=6, markerfacecolor='blue',
                markeredgecolor='blue')