python从入门到实践 第16章16.2习题答案（自己写）

# python 从入门到实践 16.2习题自练
# 比较两个地点的高温和低温 
import csv
from pathlib import Path
from datetime import datetime
import matplotlib.pyplot as plt

class value_check:
    """检查索引 表头 返回数据列表"""
    def __init__(self,path):
        self.path = Path(path)
        self.lines = self.path.read_text().splitlines()
        self.reader = csv.reader(self.lines)
    def reset_reader(self):
        """将文件指针指向文件开头"""
        self.reader = csv.reader(self.lines)
    def _title(self, show=True):
        """读完表头"""
        self.reset_reader()
        header_row = next(self.reader)
        p_header_row= [[index, value] for index,value in enumerate(header_row)]
        if show:
            for value in p_header_row:
                print(value)
        return header_row
    def r_date_value(self,value_1,value_2=None,value_1_type=float,value_2_type=float):
        """返回value 的数据"""
        print("----")
        index_value_1 = self._title(False).index(value_1)# 读完第一行
        if value_2:
            index_value_2 = self._title(False).index(value_2)# 读完第一行
        index_date = self._title(False).index('DATE')

        # print(index_value)
        # print(index_date)
        dates, values_1,values_2 = [],[],[]
        for row in self.reader:
            # print(row[2])
            try:
                date = datetime.strptime(row[index_date],'%Y-%m-%d')
                value_1_current = value_1_type(row[index_value_1])
                if value_2:
                    value_2_current = value_2_type(row[index_value_2])
                # value_current = row[index_value]
            except ValueError:
                print(f"Miss data{date} ")
            else:
                dates.append(date)
                values_1.append(value_1_current)
                if value_2:
                    values_2.append(value_2_current)
        if value_2:
            return dates, values_1, values_2
        else:
            return dates, values_1
# sitka weather data
value_check_1 = value_check('weather_data/sitka_weather_2021_full.csv')
dates, highs, lows =value_check_1.r_date_value('TMAX','TMIN')
# 绘制图形
plt.style.use('seaborn')
fig, ax = plt.subplots()
ax.plot(dates,highs,color='red',alpha=0.5)
ax.plot(dates,lows,color='blue',alpha=0.5)
ax.fill_between(dates,highs,lows,color='blue',alpha=0.1)
# death valley data
value_check_1 = value_check('weather_data/death_valley_2021_full.csv')
value_check_1._title()# 打印表头
dates, highs, lows = value_check_1.r_date_value('TMAX','TMIN')
ax.plot(dates,highs,color='red',alpha=0.3)
ax.plot(dates,lows,color='blue',alpha=0.3)
ax.fill_between(dates,highs,lows,color='blue',alpha=0.3)


ax.set_xlabel("date",fontsize=16)
ax.set_title("Prcp of date",fontsize=24)
ax.set_ylabel("prcps",fontsize=16)
fig.autofmt_xdate()
plt.show()