问题描述:
最近在学习《Python编程:从入门到实践》的第16章《下载数据》的时候,利用Python读取csv文件获取到一个温度数据,利用matplotlib绘图发现纵坐标混乱,如下图所示:
从图中可以看到,纵坐标的标注很奇怪,从下往上忽大忽小,而且毫无规律可言。
我的代码如下
import csv
from matplotlib import pyplot as plt
from datetime import datetime
# filename = 'sitka_weather_07-2014.csv'
filename = "sitka_weather_2014.csv"
with open(filename) as f:
reader = csv.reader(f)
# 获取第一行
header_row = next(reader)
# 获取日期和最高温度
dates, highs = [], []
for row in reader:
highs.append(row[1])
current_date = datetime.strptime(row[0], "%Y-%m-%d")
dates.append(current_date)
fig = plt.figure(figsize=(6, 4), dpi=128)
plt.plot(dates, highs, c='red')
# 设置图形的格式
plt.title("Daily high temperatures - 2014", fontsize=12)
plt.xlabel('', fontsize=9)
# 自动旋转横坐标的标签
fig.autofmt_xdate()
plt.ylabel("Temperature (F)", fontsize=9)
plt.tick_params(axis='both', which='major', labelsize=9)
plt.show()
经过分析之后发现,csv.reader返回的对象类型是str类型,也就是说,我直接把一个字符串列表当做数字列表拿去绘图了,其实,有问题的代码代码就一行:
highs.append(row[1])
将字符串类型转换为数值类型就可以了,我这个数据是整型数据,因此只需要更改如下:
highs.append(int(row[1]))
问题解决,最终效果如下:
参考内容: