如何将颜色条添加到直方图?

时间:2015-06-03 00:16:21

标签: python matplotlib histogram colorbar

我有这样的直方图(就像普通的直方图一样): enter image description here

在我的情况下,总共有20个条形(从0到1跨越x轴),条形的颜色是根据x轴上的值定义的。

我想要的是在直方图底部添加http://wiki.scipy.org/Cookbook/Matplotlib/Show_colormaps之一的色谱,但我不知道如何添加它。

任何帮助将不胜感激!

2 个答案:

答案 0 :(得分:6)

您需要从某种形式的色彩图中指定面部的颜色,例如,如果您想要20个分页和spectral色彩映射,

nbins = 20
colors = plt.cm.spectral(np.linspace(nbins))

然后,您可以使用它来指定条形图的颜色,这可能是最简单的方法,首先获取直方图数据(使用numpy)并绘制条形图。然后,您可以将颜色条添加到底部的单独轴。

作为一个最小的例子,

import numpy as np
import matplotlib.pyplot as plt
import matplotlib as mpl

nbins = 20
minbin = 0.
maxbin = 1.

data = np.random.normal(size=10000)
bins = np.linspace(minbin,maxbin,20)

cmap = plt.cm.spectral
norm = mpl.colors.Normalize(vmin=data.min(), vmax=data.max())
colors = cmap(bins)

hist, bin_edges = np.histogram(data, bins)

fig = plt.figure()
ax = fig.add_axes([0.05, 0.2, 0.9, 0.7])
ax1 = fig.add_axes([0.05, 0.05, 0.9, 0.1])

cb1 = mpl.colorbar.ColorbarBase(ax1, cmap=cmap,
                                norm=norm,
                                orientation='horizontal')

ax.bar(bin_edges[:-1], hist, width=0.051, color=colors, alpha=0.8)
ax.set_xlim((0., 1.))
plt.show()

哪个收益率, enter image description here

答案 1 :(得分:0)

我没有足够的声誉评论Ed Smith的答案(上图)......

......但在第14行,我认为应该是:

colors = cmap(np.linspace(minbin,maxbin,bins*maxbin))

在他的例子中一切正常,因为maxbin碰巧是1,但在其他情况下,直方图和颜色条的颜色不对齐