根据数据阵列为直方图创建色彩图

Question

我已经在各种各样的地方看过如何做到这一点，但是如果可能的话，我们无法找到我需要的东西/找到验证。

我在y轴上有一个2d的高度直方图（由rdata1给出）和在x轴上的强度（由强度给出）。直方图很好，但是我想缩放颜色条而不是标准化。

我已经定义了一个colourmap，我只是想用我绘制的数据来缩放它。

我想缩放数据，以便颜色条（当前从0-1开始）从0扩展到（bin中的点数）/（len（时间））。这样我就能找出一个点在某个高度处于特定强度区间的概率。我不确定是否通过制作第二个直方图（我不会绘制）并从中导出最大/最小值以缩放颜色条，或者使用多个for循环和列表将值附加到容器中来执行此操作对于每个高度范围，然后最大/最小值这些箱中的值的数量。

代码如下：

import numpy as np
import file_reader as fr
import matplotlib.pyplot as plt
from matplotlib.colors import LinearSegmentedColormap
import matplotlib as mpl

time = [0.01649999991059303, 0.02584999985992908]
rdata = [-600.020751953125, -570.04150390625, -540.062255859375, -510.0830078125, -480.1037902832031, -450.1245422363281, -420.1452941894531, -390.1660461425781, -360.1867980957031, -330.2075500488281]
intensity = [[-37.32464981079102, -38.3233528137207], [-37.70231628417969, -38.05134201049805], [-38.27889251708984, -38.82979583740234], [-28.01022720336914, -27.68825912475586], [-8.408446311950684, -8.440451622009277], [-8.749446868896484, -8.750232696533203], [-9.431790351867676, -9.41820240020752], [-10.09048461914062, -10.23848724365234], [-10.84317588806152, -10.84869194030762], [-11.61933135986328, -11.67543029785156]]

range_bins = np.linspace(rdata[0],rdata[-1],(len(rdata)+1))
intensity_bins = np.linspace(-70,30,100)

intensity = np.array(intensity).ravel()
rdata1 = np.repeat(rdata,len(time))

cdict = {'red': ((0.0, 1.0, 1.0),
                 (0.25, 0.0, 0.0),
                 (0.55, 0.35, 0.35),
                 (0.75, 0.75, 0.75),
                 (1.0, 1.0, 1.0)),

       'green': ((0.0, 1.0, 1.0),
                 (0.25, 0.1, 0.1),
                 (0.55, 0.6, 0.6),
                 (0.75, 0.8, 0.8),
                 (1.0, 0.0, 0.0)),

        'blue': ((0.0, 1.0, 1.0),
                 (0.25, 1.0, 1.0),
                 (0.55, 0.2, 0.2),
                 (0.75, 0.1, 0.1),
                 (1.0, 0.0, 0.0))
        }

radar_map = LinearSegmentedColormap('radar_map', cdict)

H, range_bins, intensity_bins = np.histogram2d(rdata1,intensity,bins=(range_bins,intensity_bins))

fig = plt.figure()
X,Y = np.meshgrid(intensity_bins,range_bins)
plt.pcolormesh(X,Y,H, cmap=radar_map)

cax = fig.add_axes([0.95, 0.2, 0.02, 0.6])
cb = mpl.colorbar.ColorbarBase(cax, cmap=radar_map, spacing='proportional')

对此的任何帮助都将受到广泛赞赏。对不起这篇相当冗长的帖子感到抱歉。

Answer 1

目前，颜色条与显示的pcolormesh无关。 如果将颜色条链接到pcolormesh，它将自动缩放到绘图的最小和最大级别。使用@dataprovider，其中@test是pcolormesh的返回值。

要标准化直方图计数，您可以将直方图除以您想要的任何数量，例如

plt.colorbar(pc)

示例：

pc