使用 matplotlib 繪制帶日期的坐標軸
效果圖
代碼
import numpy as np
import matplotlib.pyplot as plt
import matplotlib.dates as mdates
fig, ax = plt.subplots()
"""生成數據"""
beginDate = '2012-01-01'
endDate = '2018-01-01'
# 將日期字符串轉化為數字（從1970-01-01算起的天數差）
x = np.arange(mdates.datestr2num(beginDate), mdates.datestr2num(endDate), 15)
y = np.random.randn(len(x))*3+2
data = {}
# 將數字（天數差）轉為日期對象 numpy.datetime64
data['date'] = [np.datetime64(int(c), 'D') for c in x]
data['value'] = y
ax.plot('date', 'value', data=data)
# ax.plot(data['date'], data['value']) # 與前一行是等效的
"""設置坐標軸的格式"""
# 設置主刻度, 每6個月一個刻度
fmt_half_year = mdates.MonthLocator(interval=6)
ax.xaxis.set_major_locator(fmt_half_year)
# 設置次刻度，每個月一個刻度
fmt_month = mdates.MonthLocator() # 默認即可
ax.xaxis.set_minor_locator(fmt_month)
# 設置 x 坐標軸的刻度格式
ax.xaxis.set_major_formatter(mdates.DateFormatter("%Y-%m"))
# 設置橫坐標軸的范圍
datemin = np.datetime64(data['date'][0], 'Y')
datemax = np.datetime64(data['date'][-1], 'Y') + np.timedelta64(1, 'Y')
ax.set_xlim(datemin, datemax)
# 設置刻度的顯示格式
ax.format_xdata = mdates.DateFormatter('%Y-%m')
ax.format_ydata = lambda x : f'$x:.2f$'
ax.grid(True)
"""自動調整刻度字符串"""
# 自動調整 x 軸的刻度字符串（旋轉）使得每個字符串有足夠的空間而不重疊
fig.autofmt_xdate()
plt.show()
代碼中使用到的類簡單介紹一下，具體參數或用法可以點擊查看。
- matplotlib.dates.datestr2num() 將日期轉化為天數差
- numpy.datetime64() 將數字（天數差）轉為日期對象
numpy.datetime64 - matplotlib.dates.MonthLocator() 配合設置日期刻度間隔
- matplotlib.dates.DateFormatter() 設置日期顯示格式
- fig.autofmt_xdate() 自動調整坐標軸，未調用字符串會重疊在一起
