本文目的- 介绍了如何从nc文件中,提取风速数据;
- 介绍如何将风速数据转换成时间序列;
- 简单的时间序列的趋势拆解(首发)。- P! T) B* E* S9 [) m
9 `" R& ], H' t2 s9 n& o3 K& F; E 代码链接代码我已经放在Github上面了,免费分享使用,https://github.com/yuanzhoulvpi2 ... ree/main/python_GIS。 ' @) k# M# E& C8 w
过程介绍
8 C- e; L$ u! [) K U$ x- C' f
& p+ q3 Q7 T0 j5 {. H( k$ _; d % X3 v- e# f. `; V: N T
1. 导入包
6 Y# `8 {2 e+ ^! B- _
4 x' z9 c/ p9 R[Python] 纯文本查看 复制代码 # 基础的数据处理工具
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt # 可视化
import datetime # 处理python时间函数
import netCDF4 as nc # 处理nc数据
from netCDF4 import num2date # 处理nc数据
import geopandas as gpd # 处理网格数据,shp之类的
import rasterio # 处理tiff文件
from shapely.geometry import Point # gis的一些逻辑判断
from cartopy import crs as ccrs # 设置投影坐标系等
from tqdm import tqdm # 打印进度条
from joblib import Parallel, delayed # 并行
import platform # 检测系统
tqdm.pandas()
# matplotlib 显示中文的问题
if platform.system() == 'Darwin':
plt.rcParams["font.family"] = 'Arial Unicode MS'
elif platform.system() == 'Windows':
plt.rcParams["font.family"] = 'SimHei'
else:
pass
# q8 O' D" h8 ]* G0 V
8 W. c) X6 J* n! E+ g1 j
7 I3 y3 h/ Z$ ^* O) @3 T* z2.导入数据 处理数据# b- F {2 H/ M- @! E
_0 z% F( ^7 K
4 n# K- Z' S U0 n' Q: A* f[Python] 纯文本查看 复制代码 # 导入数据
nc_data = nc.Dataset("./数据集/GIS实践3/2016_2020.nc")
# 处理数据
raw_latitude = np.array(nc_data.variables['latitude'])
raw_longitude = np.array(nc_data.variables['longitude'])
raw_time = np.array(nc_data.variables['time'])
raw_u10 = np.array(nc_data.variables['u10'])
raw_v10 = np.array(nc_data.variables['v10'])
# 提取缺失值,并且将缺失值替换
missing_u10_value = nc_data.variables['u10'].missing_value
missing_v10_value = nc_data.variables['v10'].missing_value
raw_v10[raw_v10 == missing_v10_value] = np.nan
raw_u10[raw_u10 == missing_u10_value] = np.nan
# 处理时间
def cftime2datetime(cftime, units, format='%Y-%m-%d %H:%M:%S'):
"""
将nc文件里面的时间格式 从cftime 转换到 datetime格式
:param cftime:
:param units:
:param format:
:return:
"""
return datetime.datetime.strptime(num2date(times=cftime, units=units).strftime(format), format)
clean_time_data = pd.Series([cftime2datetime(i, units=str(nc_data.variables['time'].units)) for i in tqdm(raw_time)])
clean_time_data[:4]
0 T9 @7 d- w- K) o: P
/ Q+ o; x1 `1 ~9 ?! z! f/ L3. 计算风速数据2 D/ m; R& M, N
. _6 y6 W+ q- H4 { {& V
; i/ e# ? w' x0 P* y G3 d
[Python] 纯文本查看 复制代码 windspeed_mean = pd.Series([np.sqrt(raw_v10[i,:, :] ** 2 + raw_u10[i, :, :]**2).mean() for i in tqdm(range(clean_time_data.shape[0]))])
time_windspeed = pd.DataFrame({'time':clean_time_data,'mean_ws':windspeed_mean})
time_windspeed 7 T$ I7 O2 x) }6 J+ Z
' V# y8 W1 S7 b. {: X$ F
' W7 P0 h3 F3 q, j3 b; q& P) Z+ @& g0 r( q
4. 年度数据可视化
2 b w# L/ i3 y7 ]7 y& R6 [3 t2 v% I1 A
- j& X7 M* o8 n; X) \# T7 P[Python] 纯文本查看 复制代码 year_data = time_windspeed.groupby(time_windspeed.time.dt.year).agg(
mean_ws = ('mean_ws', 'mean')
).reset_index()
# year_data
with plt.style.context('fivethirtyeight') as style:
fig, ax = plt.subplots(figsize=(10,3), dpi=300)
ax.plot(year_data['time'], year_data['mean_ws'], '-o',linewidth=3, ms=6)
ax.set_xticks(year_data['time'])
#
#
for i in range(year_data.shape[0]):
ax.text(year_data.iloc[/size][/font][i][font=新宋体][size=3]['time']+0.1, year_data.iloc[/size][/font][i][font=新宋体][size=3]['mean_ws'], str(np.around(year_data.iloc[/size][/font][i][font=新宋体][size=3]['mean_ws'], 2)),
bbox=dict(boxstyle='round', facecolor='white', alpha=0.5))
#
for i in ['top', 'right']:
ax.spines[/size][/font][i][font=新宋体][size=3].set_visible(False)
ax.set_title("各年平均风速")
ax.set_ylabel("$Wind Speed / m.s^{-1}$")
( Z/ {* E4 i! H# D5 R
8 J& H0 \8 K$ \ j6 }+ A$ J& ]) w
. @8 p1 |- J R$ s4 R1 J
_7 T4 m: e" J& L5. 月维度数据可视化
0 Q: W4 \9 e! s: p# o) \* e[Python] 纯文本查看 复制代码 month_data = time_windspeed.groupby(time_windspeed.time.dt.month).agg(
mean_ws = ('mean_ws', 'mean')
).reset_index()
with plt.style.context('fivethirtyeight') as style:
fig, ax = plt.subplots(figsize=(10,3), dpi=300)
ax.plot(month_data['time'], month_data['mean_ws'], '-o',linewidth=3, ms=6)
ax.set_xticks(month_data['time'])
_ = ax.set_xticklabels(labels=[f'{i}月' for i in month_data['time']])
for i in range(month_data.shape[0]):
ax.text(month_data.iloc[/size][/font][i][font=新宋体][size=3]['time'], month_data.iloc[/size][/font][i][font=新宋体][size=3]['mean_ws']+0.05, str(np.around(month_data.iloc[/size][/font][i][font=新宋体][size=3]['mean_ws'], 2)),
bbox=dict(boxstyle='round', facecolor='white', alpha=0.5))
for i in ['top', 'right']:
ax.spines[/size][/font][i][font=新宋体][size=3].set_visible(False)
ax.set_title("各月平均风速")
ax.set_ylabel("$Wind Speed / m.s^{-1}$")
fig.savefig("month_plot.png") 5 y& O# R: N" j0 @% P) Z8 ^
7 ^6 l1 {2 _, G. e5 i" l K( `
! h# ?- F b6 ?8 _. T
1 J$ X& P1 S& n4 ]/ z B6.天维度数据可视化
0 r& _$ D8 Y1 }5 V- 计算天数据) f2 o4 H* _( \" I7 M/ r. C
( F2 A2 M2 L1 F! D
[Python] 纯文本查看 复制代码 day_data = time_windspeed.groupby(time_windspeed.time.apply(lambda x: x.strftime('%Y-%m-%d'))).agg(
mean_ws = ('mean_ws', 'mean')
).reset_index()
day_data['time'] = pd.to_datetime(day_data['time'])
day_data = day_data.set_index('time')
day_data.head()- 可视化
1 I4 t+ U9 ]4 m# }5 Y
7 Y4 ~) x: e8 A5 z; g" x
[Python] 纯文本查看 复制代码 # day_data.dtypes
fig, ax = plt.subplots(figsize=(20,4), dpi=300)
ax.plot(day_data.index, day_data['mean_ws'], '-o')
# ax.xaxis.set_ticks_position('none')
# ax.tick_params(axis="x", labelbottom=False)
ax.set_title("每天平均风速")
ax.set_ylabel("$Wind Speed / m.s^{-1}$")
ax.set_xlabel("date")
fig.savefig('day_plot.png') 1 z, M+ F+ }6 Q) t" f' G
( t! q1 e! J: _9 Q! j+ @! F
4 w7 L% k( k2 p9 `, ]! v
: V& Z9 x) K6 x- J
1.天维度数据做趋势拆解( j* \9 U4 ]8 y2 P3 g, J
/ h% z# Z9 b9 ?( y& M0 R" j
[Python] 纯文本查看 复制代码 # 导入包
from statsmodels.tsa.seasonal import seasonal_decompose
from dateutil.parser import parse
# 乘法模型
result_mul = seasonal_decompose(day_data['mean_ws'], model="multilicative", extrapolate_trend='freq')
result_add = seasonal_decompose(day_data['mean_ws'], model="additive", extrapolate_trend='freq')
font = {'family': 'serif',
'color': 'darkred',
'weight': 'normal',
'size': 16,
}
# 画图
with plt.style.context('classic'):
fig, ax = plt.subplots(ncols=2, nrows=4, figsize=(22, 15), sharex=True, dpi=300)
def plot_decompose(result, ax, index, title, fontdict=font):
ax[0, index].set_title(title, fontdict=fontdict)
result.observed.plot(ax=ax[0, index])
ax[0, index].set_ylabel("Observed")
result.trend.plot(ax=ax[1, index])
ax[1, index].set_ylabel("Trend")
result.seasonal.plot(ax=ax[2, index])
ax[2, index].set_ylabel("Seasonal")
result.resid.plot(ax=ax[3, index])
ax[3, index].set_ylabel("Resid")
plot_decompose(result=result_add, ax=ax, index=0, title="Additive Decompose", fontdict=font)
plot_decompose(result=result_mul, ax=ax, index=1, title="Multiplicative Decompose", fontdict=font)
fig.savefig('decompose.png')
0 }1 ~- Y3 n. c# \/ O
. G/ x& O8 ?4 R3 T$ I5 w, Y0 Q ; Z* v8 u" Q, H& y# p
9 I) L6 G' M6 H8 P: b: Y1 W
. t5 T1 f6 L: f$ i
@$ l- Z3 X/ R# Z. X0 Q |