如何向直方图添加颜色条?

问题描述 投票:0回答:1

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

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

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

如有任何帮助,我们将不胜感激!

python matplotlib histogram colorbar
1个回答
7
投票

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

其产量, enter image description here

最新问题
© www.soinside.com 2019 - 2025. All rights reserved.