打开csv文件，并使用matplotlib绘制时间趋势图

weixin_47209234

已于 2024-03-13 14:56:49 修改

阅读量986

点赞数 21

文章标签： python 信息可视化开发语言

于 2024-03-13 14:50:43 首次发布

本文链接：https://blog.csdn.net/weixin_47209234/article/details/136670301

版权

使用Matplotlib根据下载的数据创建一个图表，展示两个不同地区的温度变化：阿拉斯加州锡特卡和加利福尼亚州死亡谷。

一、总体代码

import csv
from datetime import datetime
import matplotlib.pyplot as plt

# 导入数据文件
filename1 = 'chap16/sitka_weather_2021_full.csv'
filename2 = 'chap16/death_valley_2021_full.csv'
with open(filename1) as f:
    reader = csv.reader(f)
    header_row = next(reader)
    print(header_row)


    # 确定列索引 
    for index,column_header in enumerate(header_row):
        print(index,column_header)     

    # 从文件中提取日期、温度PRCP
    dates, tmaxs ,tmins = [], [], []
    for row in reader:
        date = datetime.strptime(row[2],'%Y-%m-%d')
        try:
            tmax = int(row[7])
            tmin = int(row[8])
        except ValueError:
            print(f'{date}数据缺失！')
        else:
            tmaxs.append(tmax)
            tmins.append(tmin)
            dates.append(date)

# 绘制图形
plt.style.use('seaborn-v0_8')
fig, ax = plt.subplots()
ax.plot(dates, tmaxs, c='red',alpha=0.5)
ax.plot(dates, tmins, c='blue',alpha=0.5 )  # alpha 代表颜色
ax.fill_between(dates,tmaxs,tmins,facecolor='blue',alpha=0.1)

# 设置标签
ax.set_title('2021年温度',fontsize=24) 
fig.autofmt_xdate()
ax.set_ylabel('temperature(F)')
ax.set_ylim(0, 110)
ax.tick_params(axis='both', labelcolor='b',labelsize='medium', width=3)

plt.show()

实现效果：

二、分部分说明

0、调用并读取文件、提取数据.

①模块csv 包含函数next() ，调用它并传入阅读器对象时，它将返回文件中的下一行。只调用了next() 一次，则得到文件第一行的列名。

②遍历文件中余下的各行：for row in reader:

with open(filename) as f:
    reader = csv.reader(f)
    header_row = next(reader)
    print(header_row)


    # 确定列索引 
    for index,column_header in enumerate(header_row):
        print(index,column_header)     

    # 从文件中提取日期、温度
    dates, tmaxs ,tmins = [], [], []
    for row in reader:
        date = datetime.strptime(row[2],'%Y-%m-%d')
        try:
            tmax = int(row[7])
            tmin = int(row[8])
        except ValueError:
            print(f'{date}数据缺失！')
        else:
            tmaxs.append(tmax)
            tmins.append(tmin)
            dates.append(date)

1、绘制图形。

subplots函数可在一张图片中绘制一个或多个图表。变量fig 表示整张图片。变量ax表示图片中的各个图表。ax.plot代表绘制图标，调用两次代表在fig上绘制两张。


# 绘制图形
plt.style.use('seaborn-v0_8')
fig, ax = plt.subplots()

# 可以在一张图上绘制两条折线

ax.plot(dates, tmaxs, c='red',alpha=0.5)
ax.plot(dates, tmins, c='blue',alpha=0.5 )  # alpha 代表颜色

# 在两条线中填充
ax.fill_between(dates,tmaxs,tmins,facecolor='blue',alpha=0.1)

2、设置标签：使用ax.set_***命令。

ax.set_title('2017年每日降水量',fontsize=24) 
ax.set_xlabel('')
fig.autofmt_xdate()  // 设置倾斜
ax.set_ylabel('降水量（mm）')
ax.set_ylim(0, 2.5)  //设置固定y刻度

4、更改刻度、刻度标签和网格线的外观：包括axis（应用参数的轴）,labelcolor（标签颜色）,labelsize(标签字体大小)。

ax.tick_params(axis='both', labelcolor='b',labelsize='medium', width=3)

附录：

0、必须把当前的图关掉才能继续画图。

1、使用with打开文件，并且将其内容存储为reader变量后，后续的操作需要在with底下（即缩进）

with open(filename) as f:
    reader = csv.reader(f)
    header_row = next(reader)
    print(header_row)

    # 从文件中提取日期、每日降水量PRCP
    dates, PRCPs = [], []
    for row in reader:
        PRCP = float(row[5])
        date = datetime.strptime(row[2],'%Y-%m-%d')
        PRCPs.append(PRCP)
        dates.append(date)

2、index()方法返回列表中特定元素的索引。

# 例如
animals = ['cat', 'dog', 'mouse', 'elephant'] 
animals.index('dog')

# 定义
date_index = header_row.index('DATE') 
high_index = header_row.index('TMAX') 
low_index = header_row.index('TMIN') 
name_index = header_row.index('NAME')

3、try-except代码块：即便出现异常，程序也将继续运行：显示你编写的友好的错误消息，而不是令用户迷惑的traceback。例如以下代码：当第二个数为0时，输出提示性信息，并且再次循环运行程序。