配对时更改箱图的颜色

时间:2015-09-28 05:16:44

标签: python python-3.x matplotlib boxplot

我想根据它们代表的内容更改箱图的颜色,这是成对分组的,所以我的问题是:

如何在配对时更改箱图的颜色? 考虑到每对的第一个箱图应为蓝色,第二个箱图为红色。

这是代码,对不起,如果它很乱:

def obtenerBoxplotsAnuales(self, directorioEntrada, directorioSalida):
    meses = ["Enero","Febrero","Marzo","Abril","Mayo","Junio", "Julio", "Agosto","Septie.","Octubre","Noviem.","Diciem."]
    ciudades = ["CO","CR"]      
    anios = ["2011", "2012", "2013"]
    boxPlotMensual = []
    fig = plt.figure()
    fig.set_size_inches(14.3, 9)
    ax = plt.axes()
    plt.hold(True)
    for anio in anios:
        boxPlotAnual = []
        i=0
        ticks = []
        for mes in range(len(meses)):
            data1 = getSomeData()
            data2 = getSomeData()
            data = [    [int(float(data1[2])), int(float(data1[0])), int(float(data1[1]))],
                        [int(float(data2[2])), int(float(data2[0])), int(float(data2[1]))] ]
            plt.boxplot(data, positions=[i,i+1], widths=0.5)
            ticks.append(i+0.5)
            i=i+2
        hB, = plt.plot([1,1],'b-')
        hR, = plt.plot([1,1],'r-')
        plt.legend((hB, hR),('Caleta', 'Comodoro'))
        hB.set_visible(False)
        hR.set_visible(False)
        ax.set_xticklabels(meses)
        ax.set_xticks(ticks)
        plt.savefig(directorioSalida+"/asdasd"+str(anio)+".ps", orientation='landscape', papertype='A4' )

这是我得到的: Boxplots are all the same colour

我已经读过这个解决方案与plt.boxplot(...)返回一种包含所创建的行列表的dict对象有关,因此修改每个boxplot颜色的方法是访问索引?这个案子怎么样?

1 个答案:

答案 0 :(得分:2)

您可以从boxplot设置返回字典的颜色,如下所示,

import matplotlib.pyplot as plt
import numpy as np

nboxes = 10

# fake up some data
spread= np.random.rand(50,nboxes) * 100
center = np.ones((25,nboxes)) * 50
flier_high = np.random.rand(10,nboxes) * 100 + 100
flier_low = np.random.rand(10,nboxes) * -100
data =np.concatenate((spread, center, flier_high, flier_low), 0)

# plot figure
plt.figure()
bp = plt.boxplot(data)

for i, box in enumerate(bp['boxes']):
    #Colour alternate boxes blue and red
    if i%2:
        box.set_color('blue')
    else:
        box.set_color('red')

plt.show()

bp['boxes']中遍历所有框并使用方法set_color(您还可以box.set_markerfacecolor和其他标准matplotlib艺术家属性)的位置。 bp dict还包含['boxes', 'fliers', 'medians', 'means', 'whiskers', 'caps'],也可以根据需要进行更改。