代码:读取csv文件,日期列为“年/月/日”,以节日名称(元旦,春节,劳动,国庆)为x轴,以平均需求量为y轴,绘制2015年节假日平均需求量折线与2015年年平均需求量的折线图
时间: 2024-05-12 20:16:44 浏览: 124
谢谢您的问题,以下是代码实现:
```
import pandas as pd
import matplotlib.pyplot as plt
# 读取csv文件
data = pd.read_csv('data.csv')
# 将日期列转为datetime类型
data['日期'] = pd.to_datetime(data['日期'], format='%Y/%m/%d')
# 提取出年份和节日名称
data['年份'] = data['日期'].dt.year
data['节日'] = data['节日'].str.extract(r'(元旦|春节|劳动|国庆)')
# 选取2015年的数据
data_2015 = data[data['年份'] == 2015]
# 计算各节日的平均需求量
avg_demand_by_festival = data_2015.groupby('节日')['需求量'].mean()
# 计算2015年的年平均需求量
avg_demand_by_year = data_2015['需求量'].mean()
# 绘制折线图
x = avg_demand_by_festival.index # x轴为节日名称
y1 = avg_demand_by_festival.values # y轴为节日平均需求量
y2 = [avg_demand_by_year] * len(x) # 年平均需求量
plt.plot(x, y1, label='Festival Demand')
plt.plot(x, y2, label='Annual Demand')
plt.xlabel('Festival')
plt.ylabel('Demand')
plt.title('Demand Comparison for 2015')
plt.legend()
plt.show()
```
希望这个代码可以帮到您。如果还有其他问题,请随时问我。
阅读全文