python - matplotlib 中 Windrose 的子图

Question

我正在尝试用 4 个 windrose 子图制作一个图形，但我意识到 windrose 只有这样的轴：ax = WindroseAxes.from_ax()那么，我怎样才能用 windrose 绘制子图？

Answer 1

有两种解决方案：

(a) 从矩形创建轴

首先这里已经有一个类似的问题：How to add specific axes to matplotlib subplot?

在那里，解决方案是在图中创建一个rect带有新子图轴坐标的矩形，然后调用ax = WindroseAxes(fig, rect)

一个更容易理解的例子是

from windrose import WindroseAxes
from matplotlib import pyplot as plt
import numpy as np
ws = np.random.random(500) * 6
wd = np.random.random(500) * 360

fig=plt.figure()
rect=[0.5,0.5,0.4,0.4] 
wa=WindroseAxes(fig, rect)
fig.add_axes(wa)
wa.bar(wd, ws, normed=True, opening=0.8, edgecolor='white')

plt.show()

(b) 添加投影

现在创建这个矩形可能会很烦人，如果能够使用 matplotlib 子图功能会更好。这里
提出的一个建议是将 注册为 matplotlib 的投影。为此，您需要编辑 site-packages/windrose 中的文件 windrose.py 如下：WindroseAxes

1. from matplotlib.projections import register_projection在文件开头包含一个导入。

2. 然后添加一个名称变量：

   class WindroseAxes(PolarAxes):
       name = 'windrose'
       ...
   

3. 最后，在 windrose.py 的末尾，添加：

   register_projection(WindroseAxes)
   

完成后，您可以使用 matplotlib 轴的投影参数轻松创建 windrose 轴：

from matplotlib import pyplot as plt
import windrose
import matplotlib.cm as cm
import numpy as np

ws = np.random.random(500) * 6
wd = np.random.random(500) * 360

fig = plt.figure()
ax = fig.add_subplot(221, projection="windrose")

ax.contourf(wd, ws, bins=np.arange(0, 8, 1), cmap=cm.hot)

ax.legend(bbox_to_anchor=(1.02, 0))
plt.show()

Answer 2

要使子图具有相同的比例（例如，对于每月数据），只需在add_subplot函数中添加rmax参数。对我来说工作：

ax = fig.add_subplot(nrows, ncols, month, projection="windrose", rmax = 50)