Barplot根据色图进行着色?

时间:2017-03-07 19:21:53

标签: python matplotlib plot seaborn

首先,我对Matplotlib或Seaborn的颜色很陌生。我的目的是创建一个条形图,其条形图根据自定义调色板着色。这样的东西,但我的自定义调色板(见下面,红色,橙色,绿色和蓝色的调色板):

barplot

我使用LinearSegmentedColormap方法创建了自定义顺序调色板,但我无法在简单的plt.barplot()中使用它。当然不难,但我看不出路。我使用下面的函数创建了调色板,从这个线程获得:Create own colormap using matplotlib and plot color scale

def make_colormap(seq):
"""Return a LinearSegmentedColormap
seq: a sequence of floats and RGB-tuples. The floats should be increasing
and in the interval (0,1).
"""
seq = [(None,) * 3, 0.0] + list(seq) + [1.0, (None,) * 3]
cdict = {'red': [], 'green': [], 'blue': []}
for i, item in enumerate(seq):
    if isinstance(item, float):
        r1, g1, b1 = seq[i - 1]
        r2, g2, b2 = seq[i + 1]
        cdict['red'].append([item, r1, r2])
        cdict['green'].append([item, g1, g2])
        cdict['blue'].append([item, b1, b2])

return mcolors.LinearSegmentedColormap('CustomMap', cdict)

#main#
c = mcolors.ColorConverter().to_rgb
rvb = make_colormap(
[c('red'), 0.125, c('red'), c('orange'), 0.25, c('orange'),c('green'),0.5, c('green'),0.7, c('green'), c('blue'), 0.75, c('blue')])

N = 1000
array_dg = np.random.uniform(0, 10, size=(N, 2))
colors = np.random.uniform(0, 5, size=(N,))
plt.scatter(array_dg[:, 0], array_dg[:, 1], c=colors, cmap=rvb)
plt.colorbar()
plt.show()

返回此图:

plot palette ok

据我所知,我不能使用色图(来自LinearSegmentedColormap()的对象类型?)作为条形图,但色彩图是我获得自定义顺序调色板的独特方式。

总之,我想将第二个图(散点图)的色彩图应用到第一个图(条形图)。现在我不能这样做,因为barplot()函数没有接受LinearSegmentedColormap对象类型的参数。

我可能会比现实更难,所以我会感谢任何更清洁或更正确的方式。

2 个答案:

答案 0 :(得分:7)

要获得条形图,其中条形图根据色彩图进行着色,您可以使用color的{​​{1}}参数,其中bar(x,y, color=colors)是条形长度列表,包含所有条形图颜色。即该列表中的colors条目是i条的颜色 要从色彩映射创建此列表,您需要使用相应的值调用色彩映射。

enter image description here

i

答案 1 :(得分:1)

Seaborn colspan="2"对此非常有用,example

barplot