编写代码:如何用python读取csv,提取日期中的日和年,筛选年份中的2015年,并计算每日产品总需求量,然后绘制2015年的每日产品需求量变化曲线。x轴为日(一年365天),y轴为每日需求量,表头为2015年每日需求量变化曲线。
时间: 2024-06-10 12:05:46 浏览: 63
python 读取目录下csv文件并绘制曲线v111的方法
好的,针对您的问题,我可以给您提供以下的代码实现:
```python
import csv
from datetime import datetime
# 读取csv文件并提取日期中的日和年
with open('yourfile.csv', 'r') as csvfile:
csvreader = csv.reader(csvfile)
headers = next(csvreader) # 跳过表头
dates = []
demands = []
for row in csvreader:
date = datetime.strptime(row[0], '%Y-%m-%d')
day = date.timetuple().tm_yday # 获取一年中的第几天
year = date.year
if year == 2015:
dates.append(day)
demands.append(int(row[1]))
# 计算2015年每日产品总需求量
total_demands = [0] * 365 # 先初始化为全0
for i in range(len(dates)):
total_demands[dates[i]-1] += demands[i]
# 绘制2015年每日产品需求量变化曲线
import matplotlib.pyplot as plt
plt.plot(range(1, 366), total_demands)
plt.xlabel('Day')
plt.xticks(range(1, 366, 30), [str(datetime(2015, 1, i).date()) for i in range(1, 32, 2)])
plt.ylabel('Demand')
plt.title('2015 Daily Demand Curve')
plt.show()
```
您可以将代码中的`yourfile.csv`替换成您的CSV文件名,并且根据需要进行修改调整。希望能帮到您。
阅读全文