csv的日期、降水、流量数据绘图

import pandas as pd
import matplotlib.pyplot as plt
import numpy as np

# 读取CSV文件并转换为DataFrame格式
data = pd.read_csv('1010070accurate.csv', parse_dates=['date'])
#data= data.iloc[:]

#提取不同年份
# 按照年份分组
groups = data.groupby(data['date'].dt.year)

# 将每个组的数据存入二维数组中
data1 = []
for year, group in groups:
    arr = group[['date', 'precipitation', 'flows']]#加.values的后缀表示只保留变量不保留列名
    data1.append(arr)

print(data1[1])

# 绘制日期-降水、日期-流量折线图

fig, ax = plt.subplots()
#ax.plot(data["date"], data["precipitation"], label="Precipitation")
ax.plot(data1[1]["date"], data1[1]["precipitation"], label="Precipitation")
ax.set_xlabel("Date")
ax.set_ylabel("Precipitation")
ax.set_title("Precipitation")
ax.legend()

fig, ax = plt.subplots()
#ax.plot(data["date"], data["flows"], label="Flows")
ax.plot(data1[1]["date"], data1[1]["flows"], label="Flows")
ax.set_xlabel("Date")
ax.set_ylabel("Flows")
ax.set_title("Flows")
ax.legend()



# 绘制降水量和流量的散点图
fig, ax = plt.subplots()
#ax.scatter(data['precipitation'][114:-2], data['flows'][115:-1])
#ax.scatter(data['precipitation'], data['flows'])
ax.scatter(data1[1]['precipitation'][0:-2],data1[1]['flows'][1:-1])
ax.set_xlabel("Precipitation")
ax.set_ylabel("Flow")
ax.set_title("Precipitation vs Flow")


# 显示图形
plt.show()