python两个数组画图_python-将两个2D数组合并为一个图像

我有两个2D数组,每个数组代表一个地图上的属性.其中一个代表冰雹发生的概率(0％-100％),另一个则代表冰雹的严重程度(0-无冰雹,1-低,2-中,3-高).

我可以使用matplotlib的imshow和预定义的颜色图分别绘制此数组：

import matplotlib.pyplot as plt

import matplotlib.colors as cl

cmap = cl.ListedColormap(['#00FF00', '#FFFF00', '#FF0000'])

bounds = [0, 30, 60, 100]

norm = cl.BoundaryNorm(bounds, cmap.N)

plt.subplot(121)

plt.imshow(hail_prob, cmap=cmap, norm=norm)

cmap = cl.ListedColormap(['#00FF00', '#FFFF00', '#FF0000'])

bounds = [0.5, 1.5, 2.5, 3.5]

norm = cl.BoundaryNorm(bounds, cmap.N)

plt.subplot(122)

plt.imshow(hail_sev, cmap=cmap, norm=norm)

如上所示,这非常容易.

但是我想要一个结合了这两个功能的独特情节.我已经测试了轮廓函数,但是数据非常不规则,并且绘图看起来很糟糕.

我一直在考虑将两个特征组合到一个颜色图中,但是我不确定如何做到这一点.假设我希望每种概率和严重性组合都使用一种颜色.

有关如何执行此操作的任何想法？

解决方法:

我会做一个散点图,其中颜色是一个值,大小是另一值.例如,颜色可以是概率,但是大小可以是强度.

这是一些随机数据

hail_prob = np.random.rand(48, 64)

hail_sev = np.random.randint(0,4,hail_sev.shape)

在这里,从现有数据中,您可以使用np.meshgrid捕获x-y点并将其用于散点图中：

x = np.arange(hail_prob.shape[1])

y = np.arange(hail_prob.shape[0])

xy = np.meshgrid(x,y)

scatter(*xy, c=hail_prob, s=hail_sev)

您必须调整尺寸的归一化,因为您的单位将与良好的像素尺寸有所不同.

或更有趣的形状：

标签：matplotlib,python,numpy

来源： https://codeday.me/bug/20191122/2062350.html