matplotlib Windrose图例括号格式和位置

kadbb459  于 2022-11-15  发布在  其他
关注(0)|答案(1)|浏览(115)

下面我使用Windrose基于this绘制了一个风玫瑰图。首先,图例覆盖了玫瑰图的一部分,但当我尝试使用loc设置其位置时,图例消失了。
其次,图例的右括号是错误的,即[0.0 : 1.0[。我知道如何将其修复为[0.0 : 1.0]

密码:

from windrose import WindroseAxes
import matplotlib.cm as cm
import matplotlib.pyplot as plt
import numpy as np
import pandas as pd

df = pd.read_csv("C:\2007_GG_wind rose.csv")
ws_SAR = df[' SARwind_10m']
wd_SAR = df['wind direction SAR model_int']
ws_mde = df['gg_mde']
wd_mde = df['wind direction MDE ']

ax=WindroseAxes.from_ax()
ax.bar(wd_SAR,ws_SAR,normed=True, opening=0.8, edgecolor='white')
ax.set_legend()
plt.title("SAR 10m U",y=1.08) #y=1.08 raises the title
mklgxw1f

mklgxw1f1#

将原始的windrose.py从你的python文件夹复制到你想要的工作目录。以其他方式命名副本,例如windrose_edit.py。编辑文件并查找get_labels()函数。我是这样编辑的,但是你可以根据自己的目的采用它。

def get_labels():
        labels = np.copy(self._info['bins'])
        labels = ["%.1f : %0.1f" %(labels[i], labels[i+1]-0.1) \
                  for i in range(len(labels)-1)]
        return labels

你也可以增加图例的字体大小几行以下。

def set_legend(self):
    l = self.legend(borderaxespad=-0.10)
    plt.setp(l.get_texts(), fontsize=12)

最后导入您编辑的文件,例如import windrose_edit as wind2
并将其与

winddir_ = yourdata
windspeed_ = yourdata

fig = plt.figure(figsize=(12, 8), dpi=100, facecolor='w', edgecolor='w')
rect = [0.1, 0.1, 0.8, 0.8]

ax = wind2.WindroseAxes(fig, rect, facecolor='w')

ax.contourf(winddir_, windspeed_, bins=6, normed=True, cmap=cm.RdYlBu_r)
ax.set_legend()

虽然这是一个相对快速和懒惰的解决方案,但并不美观

相关问题