python绘制风速风向图

耿瑞满

# -*- coding: utf-8 -*-
"""
Created on Mon Mar 27 08:24:24 2023

@author: hp
"""

# 導入包
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
#-------設置支持中文----------------------#
import matplotlib as mpl
mpl.rcParams['font.sans-serif'] = ['SimHei']   #設置簡黑字體
mpl.rcParams['axes.unicode_minus'] = False
#-------自定義座標軸刻度格式----------------#
from matplotlib.ticker import FuncFormatter
np.random.seed(0)
data = pd.DataFrame(np.random.randint(20, 300, (4, 16)),
                    index=['0~0.2', '0.3~1.5', '1.6~3.3', '3.4~5.4'],
                    columns='N NNE NE ENE E ESE SE SSE S SSW SW WSW W WNW NW NNW'.split())
_sum = data.apply(np.sum)
data = data / _sum
N = 16 # 風速分佈爲16個方向
theta = np.linspace(0, 2*np.pi, N, endpoint=False) # 獲取16個方向的角度值
width = np.pi / N  # 繪製扇型的寬度,可以自行調整
labels = list(data.columns) # 自定義座標標籤爲 N , NSN, ……
# 開始繪圖
plt.figure(figsize=(8,8))
ax = plt.subplot(111, projection='polar')
for idx in data.index:
    # 每一行繪製一個扇形
    radii = data.loc[idx] # 每一行數據
    ax.bar(theta, radii, width=width, bottom=0.0, label=idx, tick_label=labels)
#------------------------------------#
ax.set_theta_zero_location('N') #設置零度方向北
ax.set_theta_direction(-1)    # 逆時針方向繪圖
#--------自定義yaxis的刻度格式-----------#
plt.gca().yaxis.set_major_formatter(FuncFormatter(lambda s, position: '{:.0f}%'.format(100*s)))
plt.title('風玫瑰圖示意圖')
plt.legend(loc=4, bbox_to_anchor=(1.15, -0.07)) # 將label顯示出來, 並調整位置
plt.savefig('E://py画图//4.png',dpi=600)
plt.show()

张一鸣

学习了