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

Question

我有一个像这样的直方图（就像普通的直方图）： 在我的情况下，总是有20个条形（跨越从0到1的x轴），条形的颜色根据x轴上的值定义。

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

任何帮助都将不胜感激！

- TYZ

---

你可以添加另一个子图。这个SO帖子中的答案将会有所帮助：https://dev59.com/VmYr5IYBdhLWcg3wTYgQ。 - WGS

请展示生成图表所使用的代码，这将帮助我们为您的用例量身定制答案。 - leekaiinthesky

Answer 1

您需要从某种颜色映射中指定面的颜色，例如，如果您想要 20 个 bin 和一个 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()

结果为：

这将产生以下结果，

- Ed Smith

---

这正是我想要的！谢谢。我会尝试将其应用于多个子图。 - TYZ

我知道我可以通过迭代调色板来在不同的x值上放置不同的颜色。如果想让调色板与y值相对应，是否有比使用反函数更简单的方法？ - user7345804