我有两个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