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

Question

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

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

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

任何帮助,将不胜感激!

Answer 1

您需要从某种形式的色彩图中指定面的颜色,例如,如果您需要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()

哪个收益率,