效果圖:
主要步驟:
1. 數據準備:下載Hadley Centre observations datasets的HadSST數據
可參考【氣象常用】時間序列圖-CSDN博客
2. 數據處理:計算線性擬合
3. 圖像繪制:繪制折線及擬合線,并添加文本
詳細代碼:著急的直接拖到最后有完整代碼
步驟一:導入庫包及圖片存儲路徑并設置中文字體為宋體,西文為新羅馬(沒有的庫包要先下好奧)
###############################################################################
# 導入庫并設置字體
import matplotlib.pyplot as plt
import numpy as np
import pandas as pd
from scipy.stats import linregress
# 設置西文字體為新羅馬字體,中文為宋體,字號為12
from matplotlib import rcParamsconfig = {"font.family": 'serif',"font.size": 12,"mathtext.fontset": 'stix',"font.serif": ['SimSun'],}
rcParams.update(config)
rcParams['axes.unicode_minus']=False
步驟二:從csv文件讀入數據
###############################################################################
datapath = r'H:/00.csdn/01data/'
figpath = r'H:/00.csdn/02fig/'# 讀取csv文件,并選擇指定的列
datan = pd.read_csv(datapath+'HadSST.4.0.1.0_annual_NHEM.csv', usecols=['year', 'anomaly', 'lower_bound_95pct_bias_uncertainty_range', 'upper_bound_95pct_bias_uncertainty_range'])xn = datan['year']
yn = datan['anomaly']
步驟三:計算擬合線
###############################################################################
# 擬合線計算
k1, b1, r_value1, p_value1, std_err1 = linregress(xn[0:51], yn[0:51])
k2, b2, r_value2, p_value2, std_err2 = linregress(xn[51:121], yn[51:121])
k3, b3, r_value3, p_value3, std_err3 = linregress(xn[121:], yn[121:])# 根據回歸系數生成擬合直線的x值
x_fit1 = np.linspace(1850, 1901, 51)
x_fit2 = np.linspace(1901, 1970, 69)
x_fit3 = np.linspace(1971, 2025, 54)# 計算擬合直線的y值
y_fit1 = k1 * x_fit1 + b1
y_fit2 = k2 * x_fit2 + b2
y_fit3 = k3 * x_fit3 + b3
步驟四:繪制圖像并添加文本
###############################################################################
#繪制圖像
fig = plt.figure(figsize=(15, 8))
ax = fig.add_axes([0.1, 0.6, 0.6, 0.4])# 繪制海溫折線
ax.plot(xn, yn, 'k-', linewidth=1, )
ax.plot(x_fit1, y_fit1, 'k--', linewidth=1, )
ax.plot(x_fit2, y_fit2, 'k--', linewidth=1, )
ax.plot(x_fit3, y_fit3, 'k--', linewidth=1, )# 添加數值
ax.text(1850, -0.6 , 'y = %7.4f * x + %7.4f' %(k1, abs(b1)), color='k', fontsize=10, rotation=0, horizontalalignment="left")
ax.text(1920, -0.6 , 'y = %7.4f * x - %7.4f' %(k2, abs(b2)), color='k', fontsize=10, rotation=0, horizontalalignment="left")
ax.text(1990, -0.6 , 'y = %7.4f * x - %7.4f' %(k3, abs(b3)), color='k', fontsize=10, rotation=0, horizontalalignment="left")ax.text(1850, -0.7 , 'r_value:%7.4f' %(r_value1), color='k', fontsize=10, rotation=0, horizontalalignment="left")
ax.text(1920, -0.7 , 'r_value:%7.4f' %(r_value2), color='k', fontsize=10, rotation=0, horizontalalignment="left")
ax.text(1990, -0.7 , 'r_value:%7.4f' %(r_value3), color='k', fontsize=10, rotation=0, horizontalalignment="left")ax.text(1850, -0.8 , 'std_err:%7.4f' %(std_err1), color='k', fontsize=10, rotation=0, horizontalalignment="left")
ax.text(1920, -0.8 , 'std_err:%7.4f' %(std_err2), color='k', fontsize=10, rotation=0, horizontalalignment="left")
ax.text(1990, -0.8 , 'std_err:%7.4f' %(std_err3), color='k', fontsize=10, rotation=0, horizontalalignment="left")# 設置坐標軸信息
# xlim,ylim:設置x軸y軸顯示得最大最小值
# xticks,yticks: 設置坐標軸刻度,可控制隔幾個標注,本圖為兩個一標
# xticklabels,yticklabels:設置坐標軸刻度名稱,維度要和xticks,yticks一致奧
# xlabel,ylabel:坐標軸名稱
# title:圖名
xlabels = np.arange(1850, 2025, 20)
ylabels = np.arange(-12, 13, 4)/10
ax.set_xticks(xlabels)
ax.set_xticklabels(xlabels, rotation = 0, fontsize = 12)
ax.set_yticks(ylabels)
ax.set_yticklabels(ylabels, fontsize = 12)
ax.set_xlabel('Year',fontsize=12)
ax.set_ylabel('Temperature Anomaly(℃)',fontsize=12)
ax.set_title('Sea Surface Temperature (HadSST4) Anomalies Over the Last 100 Years', fontsize=14)# 設置水平檢驗線
plt.axhline(0, color='grey', linestyle='--')
步驟五:保存圖像
###############################################################################
# 保存圖像
plt.savefig(figpath+'208 時間序列線性擬合.png', bbox_inches = 'tight', dpi=600, format='png')
plt.show()
完整代碼在這里:
###############################################################################
# 導入庫并設置字體
import matplotlib.pyplot as plt
import numpy as np
import pandas as pd
from scipy.stats import linregress
# 設置西文字體為新羅馬字體,中文為宋體,字號為12
from matplotlib import rcParamsconfig = {"font.family": 'serif',"font.size": 12,"mathtext.fontset": 'stix',"font.serif": ['SimSun'],}
rcParams.update(config)
rcParams['axes.unicode_minus']=False
###############################################################################
datapath = r'H:/00.csdn/01data/'
figpath = r'H:/00.csdn/02fig/'# 讀取csv文件,并選擇指定的列
datan = pd.read_csv(datapath+'HadSST.4.0.1.0_annual_NHEM.csv', usecols=['year', 'anomaly', 'lower_bound_95pct_bias_uncertainty_range', 'upper_bound_95pct_bias_uncertainty_range'])xn = datan['year']
yn = datan['anomaly']
###############################################################################
# 擬合線計算
k1, b1, r_value1, p_value1, std_err1 = linregress(xn[0:51], yn[0:51])
k2, b2, r_value2, p_value2, std_err2 = linregress(xn[51:121], yn[51:121])
k3, b3, r_value3, p_value3, std_err3 = linregress(xn[121:], yn[121:])# 根據回歸系數生成擬合直線的x值
x_fit1 = np.linspace(1850, 1901, 51)
x_fit2 = np.linspace(1901, 1970, 69)
x_fit3 = np.linspace(1971, 2025, 54)# 計算擬合直線的y值
y_fit1 = k1 * x_fit1 + b1
y_fit2 = k2 * x_fit2 + b2
y_fit3 = k3 * x_fit3 + b3
###############################################################################
#繪制圖像
fig = plt.figure(figsize=(15, 8))
ax = fig.add_axes([0.1, 0.6, 0.6, 0.4])# 繪制海溫折線
ax.plot(xn, yn, 'k-', linewidth=1, )
ax.plot(x_fit1, y_fit1, 'k--', linewidth=1, )
ax.plot(x_fit2, y_fit2, 'k--', linewidth=1, )
ax.plot(x_fit3, y_fit3, 'k--', linewidth=1, )# 添加數值
ax.text(1850, -0.6 , 'y = %7.4f * x + %7.4f' %(k1, abs(b1)), color='k', fontsize=10, rotation=0, horizontalalignment="left")
ax.text(1920, -0.6 , 'y = %7.4f * x - %7.4f' %(k2, abs(b2)), color='k', fontsize=10, rotation=0, horizontalalignment="left")
ax.text(1990, -0.6 , 'y = %7.4f * x - %7.4f' %(k3, abs(b3)), color='k', fontsize=10, rotation=0, horizontalalignment="left")ax.text(1850, -0.7 , 'r_value:%7.4f' %(r_value1), color='k', fontsize=10, rotation=0, horizontalalignment="left")
ax.text(1920, -0.7 , 'r_value:%7.4f' %(r_value2), color='k', fontsize=10, rotation=0, horizontalalignment="left")
ax.text(1990, -0.7 , 'r_value:%7.4f' %(r_value3), color='k', fontsize=10, rotation=0, horizontalalignment="left")ax.text(1850, -0.8 , 'std_err:%7.4f' %(std_err1), color='k', fontsize=10, rotation=0, horizontalalignment="left")
ax.text(1920, -0.8 , 'std_err:%7.4f' %(std_err2), color='k', fontsize=10, rotation=0, horizontalalignment="left")
ax.text(1990, -0.8 , 'std_err:%7.4f' %(std_err3), color='k', fontsize=10, rotation=0, horizontalalignment="left")# 設置坐標軸信息
# xlim,ylim:設置x軸y軸顯示得最大最小值
# xticks,yticks: 設置坐標軸刻度,可控制隔幾個標注,本圖為兩個一標
# xticklabels,yticklabels:設置坐標軸刻度名稱,維度要和xticks,yticks一致奧
# xlabel,ylabel:坐標軸名稱
# title:圖名
xlabels = np.arange(1850, 2025, 20)
ylabels = np.arange(-12, 13, 4)/10
ax.set_xticks(xlabels)
ax.set_xticklabels(xlabels, rotation = 0, fontsize = 12)
ax.set_yticks(ylabels)
ax.set_yticklabels(ylabels, fontsize = 12)
ax.set_xlabel('Year',fontsize=12)
ax.set_ylabel('Temperature Anomaly(℃)',fontsize=12)
ax.set_title('Sea Surface Temperature (HadSST4) Anomalies Over the Last 100 Years', fontsize=14)# 設置水平檢驗線
plt.axhline(0, color='grey', linestyle='--')
###############################################################################
# 保存圖像
plt.savefig(figpath+'208 時間序列線性擬合.png', bbox_inches = 'tight', dpi=600, format='png')
plt.show()