Python plt将日期横坐标设置按年显示|极客笔记

Python plt将日期横坐标设置按年显示

在使用 Python 中的 matplotlib 库进行数据可视化时，经常会遇到需要将日期作为横坐标的情况，例如绘制时间序列数据的折线图或者柱状图。但是默认情况下，matplotlib 会以日期格式显示横坐标，如果日期比较密集，会导致横坐标上的日期标签重叠，影响数据可视化效果。因此，我们可以通过设置横坐标的显示格式，将日期横坐标按照一定的时间间隔进行显示，这样可以使图表更加清晰和易读。

本文将详细介绍如何使用 matplotlib 中的 DateFormatter 和 DateLocator 类，以及 AutoDateLocator 类的设置，来将日期横坐标按年显示的方法。

准备工作

在开始之前，首先需要安装 matplotlib 库，并导入需要的模块：

import matplotlib.pyplot as plt
import matplotlib.dates as mdates
from datetime import datetime

为了演示效果，我们生成一组时间序列数据，包括日期和对应的数值，用于绘制折线图：

dates = ['2020-01-01', '2020-02-01', '2020-03-01', '2020-04-01', '2020-05-01', 
         '2020-06-01', '2020-07-01', '2020-08-01', '2020-09-01', '2020-10-01', 
         '2020-11-01', '2020-12-01']
values = [100, 120, 150, 130, 140, 160, 170, 180, 185, 190, 200, 210]

设置日期横坐标按年显示

在绘制折线图之前，我们需要设置坐标轴的时间格式。可以使用 DateFormatter 类设置日期的显示格式，使用 DateLocator 类设置日期的间隔，这样可以使日期横坐标按照一定的时间间隔进行显示。

# 将日期字符串转换为 datetime 对象
dates = [datetime.strptime(date, "%Y-%m-%d") for date in dates]

# 创建一个图形和坐标轴
fig, ax = plt.subplots()

# 绘制折线图
ax.plot(dates, values)

# 设置横坐标的日期显示格式为年份
date_format = mdates.DateFormatter('%Y')
ax.xaxis.set_major_formatter(date_format)

# 设置横坐标的日期间隔为一年
ax.xaxis.set_major_locator(mdates.YearLocator())

# 旋转日期标签，防止重叠
plt.xticks(rotation=45)

# 显示图形
plt.show()

通过以上代码，我们成功将日期横坐标按年显示，使图表更加清晰和易读。同时，日期标签会以年份的形式显示，避免了重叠现象，提高了数据可视化效果。