首页 > 解决方案 > 如何根据 python/matplotlib 中的 pandas df 列名称/标签名称为箱形图指定标签颜色

问题描述

我在子图中有一系列箱线图(见图),但想根据其 DUT1 或 DUT2 更改每个单独图的标签颜色。 在此处输入图像描述

这是用于制作箱线图的 df 样本之一。每个图都是使用类似的 df 绘制的,但包括来自不同设定点的测量值。

      DUT1 A   DUT1 B   DUT2 A   DUT2 B
527  0.92342  0.96342  0.98342  1.00342
528  0.92754  0.88754  0.97754  0.97754
529  0.93655  0.95655  0.99655  0.91655

上面的每个数字都是在指定设定点进行每次测量的 DUT 传感器和参考传感器之间的差异。

这是我如何创建箱形图的每个子图的快照。

fig, axes = plt.subplots(ncols = 4, nrows = 2, sharey = True, figsize = (10,6))
fig.add_subplot(111, frame_on = False)
plt.tick_params(labelcolor = 'none', bottom = False, left = False)
plt.ylabel('Difference from Reference PRT in $^\circ$F', labelpad=20)

boxplot1 = df_setpoint1.boxplot(column = ['DUT1 A','DUT1 B','DUT2 A','DUT2 B'], ax = axes[1,0], rot=45)
boxplot2 = df_setpoint2.boxplot(column = ['DUT1 A','DUT1 B','DUT2 A','DUT2 B'], ax = axes[1,1], rot=45)


axes[1,0].set_title('Set Point -38.5$^\circ$F')
axes[1,1].set_title('Set Point -25$^\circ$F')


fig.suptitle('Temperature Distribution for Temperature Accuracy Testing')
plt.tight_layout()
plt.show()

经过一番搜索,我在弄清楚如何为每个绘图标签指定颜色时遇到了问题,因为我不是单独创建每个箱形图,而是从每个数据框中的指定列创建。我遇到了很多自定义填充颜色框并添加了自定义颜色的图例,但我只是想控制 x 轴上文本标签的颜色。

标签: pythonpandasmatplotlib

解决方案


虽然文本标签着色不是很方便,但matplotlib后端的箱线图着色是可行的,需要在每个修补的艺术家之间循环。对于每个配对DUT1并与每个艺术家一起DUT2运行 elementwsie循环。zip

下面运行一个由 OP 提供的数据的子图。为避免重复行集成到定义的方法中,并通过它运行每个子图,或添加顶层以遍历数组for中所有生成的子图。axes

import matplotlib.patches as mpatches
...

# BOX PLOT LEGEND
blue_patch = mpatches.Patch(color='blue', label='The red data')
green_patch = mpatches.Patch(color='green', label='The blue data')
fig.legend(handles=[blue_patch, green_patch], labels=['A', 'B'], 
           ncol=2, loc='upper center', bbox_to_anchor=(0.5, 0.95))
        
# BOX PLOT      
boxplot1 = df_setpoint1.boxplot(column = ['DUT1 A','DUT1 B','DUT2 A','DUT2 B'],
                                ax=axes[1,0], patch_artist=True, rot=45)

# BOX PLOT COLORING
colors = ['blue', 'blue', 'green', 'green']

for i,(artist, col) in enumerate(zip(axes[1,0].artists, colors)):
    artist.set_edgecolor(col)
    artist.set_facecolor(col) 

    # Each box has 6 associated Line2D objects (to make the whiskers, fliers, etc.)
    # Loop over them here, and use the same colour as above
    for j in range(i*6,i*6+4):
        line = axes[1,0].lines[j]
        line.set_color(col)
        line.set_mfc(col)
        line.set_mec(col)
        line.set_linewidth(0.5)
...

fig.tight_layout(rect=[0, 0.03, 1, 0.90])
plt.show()

箱线图输出


推荐阅读