如何將特定座標軸添加到matplotlib子圖中？

Question

我想用matplotlib做一個矩陣圖。

個人情節是由一個特定的模塊windrose，其中的小類PolarAxes。然而，模塊中並沒有定義任何被稱爲subplot kwargs的投影。標準polar投影不起作用，因爲某些子類參數丟失。

我已經測試了幾種方法沒有成功（即使考慮到這個職位的海豹地圖：https://stackoverflow.com/a/25702476/3416205）。以下是我嘗試過的最接近的。如果沒有正確創建與特定WindroseAxes相關的新matplotlib投影，有什麼方法可以做我想要的嗎？

import pandas as pd 
import matplotlib.pyplot as plt 
import matplotlib.gridspec as gridspec 
from windrose import WindroseAxes 

df = pd.read_csv('https://raw.githubusercontent.com/AntoineGautier/Data/master/tmp.csv') 
fig = plt.figure() 
gs = gridspec.GridSpec(4, 2) 

def wind_plot(x, y, title=None, axes=None, fig=None): 
    ax = WindroseAxes.from_ax() 
    ax.set_position(axes.get_position(fig)) 
    ax.bar(x, y, normed=True, opening=0.8, edgecolor='white', bins=[0, 2.5, 5, 7.5, 10]) 
    ax.set_title(title) 

for (id_s, s) in enumerate(pd.unique(df.saison)): 
    for (id_jn, jn) in enumerate(pd.unique(df.jn)): 
     tmp = df.query('saison==@s & jn==@jn') 
     _ = plt.subplot(gs[id_s, id_jn], polar=True) 
     wind_plot(tmp.wd, tmp.ws, title=s + ' - ' + jn, axes=_, fig=fig) 

plt.show() 

Answer 1

從windrose模塊GitHub的頁面實際上提供次要情節的例子：https://github.com/scls19fr/windrose/blob/master/samples/example_subplots.py。

以下的作品。

import pandas as pd 
import matplotlib.pyplot as plt 
import matplotlib.gridspec as gridspec 
from windrose import WindroseAxes 

df = pd.read_csv('https://raw.githubusercontent.com/AntoineGautier/Data/master/tmp.csv') 

fig = plt.figure(figsize=(20, 10)) 
gs = gridspec.GridSpec(2, 4) 
gp = gs.get_grid_positions(fig) # [bottom, top, left, right] 

def wind_plot(x, y, title, fig, rect): 
    ax = WindroseAxes(fig, rect) 
    fig.add_axes(ax) 
    ax.bar(x, y, normed=True, opening=0.8, edgecolor='white', bins=[0, 2.5, 5, 7.5, 10]) 
    ax.set_title(title, position=(0.5, 1.1)) 

for (id_s, s) in enumerate(pd.unique(df.saison)): 
    for (id_jn, jn) in enumerate(pd.unique(df.jn)): 
     tmp = df.query('saison==@s & jn==@jn') 
     rect = [gp[2][id_s], gp[0][id_jn], 
       gp[3][id_s]-gp[2][id_s], 
       gp[1][id_jn]-gp[0][id_jn]] # [left, bottom, width, height] 
     wind_plot(tmp.wd, tmp.ws, s + ' | ' + jn, fig, rect) 

plt.show()