这篇文章主要讲解了“Python如何读取CSV文件并进行数据可视化绘图”,文中的讲解内容简单清晰,易于学习与理解,下面请大家跟着小编的思路慢慢深入,一起来研究和学习“Python如何读取CSV文件并进行数据可视化绘图”吧!
介绍:文件 sitka_weather_07-2018_simple.csv是阿拉斯加州锡特卡2018年1月1日的天气数据,其中包含当天的最高温度和最低温度。数据文件存储与data文件夹下,接下来用Python读取该文件数据,再基于数据进行可视化绘图。
sitka_highs.py
import csv # 导入csv模块from datetime import datetimeimport matplotlib.pyplot as pltfilename = 'data/sitka_weather_07-2018_simple.csv'with open(filename) as f: reader = csv.reader(f) header_row = next(reader) # 返回文件的下一行,在这便是首行,即文件头 # for index, column_header in enumerate(header_row): # 对列表调用了 enumerate()来获取每个元素的索引及其值,方便我们提取需要的数据列 # print(index, column_header) # 从文件中获取最高温度 dates, highs = [], [] for row in reader: current_date = datetime.strptime(row[2], '%Y-%m-%d') high = int(row[5]) dates.append(current_date) highs.append(high) # 根据最高温度绘制图形plt.style.use('seaborn')fig, ax = plt.subplots()ax.plot(dates, highs, c='red')# 设置图形的格式ax.set_title("2018年7月每日最高温度", fontproperties="SimHei", fontsize=24)ax.set_xlabel('', fontproperties="SimHei", fontsize=16)fig.autofmt_xdate()ax.set_ylabel("温度(F)", fontproperties="SimHei", fontsize=16)ax.tick_params(axis='both', which='major', labelsize=16)plt.show()
运行结果如下:
设置以上图标后,我们来添加更多的数据,生成一副更复杂的锡特卡天气图。将sitka_weather_2018_simple.csv数据文件置于data文件夹下,该文件包含整年的锡特卡天气数据。
对代码进行修改:
sitka_highs.py
import csv # 导入csv模块from datetime import datetimeimport matplotlib.pyplot as pltfilename = 'data/sitka_weather_2018_simple.csv'with open(filename) as f: reader = csv.reader(f) header_row = next(reader) # 返回文件的下一行,在这便是首行,即文件头 # for index, column_header in enumerate(header_row): # 对列表调用了 enumerate()来获取每个元素的索引及其值,方便我们提取需要的数据列 # print(index, column_header) # 从文件中获取最高温度 dates, highs = [], [] for row in reader: current_date = datetime.strptime(row[2], '%Y-%m-%d') high = int(row[5]) dates.append(current_date) highs.append(high) # 根据最高温度绘制图形plt.style.use('seaborn')fig, ax = plt.subplots()ax.plot(dates, highs, c='red')# 设置图形的格式ax.set_title("2018年每日最高温度", fontproperties="SimHei", fontsize=24)ax.set_xlabel('', fontproperties="SimHei", fontsize=16)fig.autofmt_xdate()ax.set_ylabel("温度(F)", fontproperties="SimHei", fontsize=16)ax.tick_params(axis='both', which='major', labelsize=16)plt.show()
运行结果如下:
代码再改进:虽然上图已经显示了丰富的数据,但是还能再添加最低温度数据,使其更有用
对代码进行修改:
sitka_highs_lows.py
import csv # 导入csv模块from datetime import datetimeimport matplotlib.pyplot as pltfilename = 'data/sitka_weather_2018_simple.csv'with open(filename) as f: reader = csv.reader(f) header_row = next(reader) # 返回文件的下一行,在这便是首行,即文件头 # for index, column_header in enumerate(header_row): # 对列表调用了 enumerate()来获取每个元素的索引及其值,方便我们提取需要的数据列 # print(index, column_header) # 从文件中获取日期、最高温度和最低温度 dates, highs, lows = [], [], [] for row in reader: current_date = datetime.strptime(row[2], '%Y-%m-%d') high = int(row[5]) low = int(row[6]) dates.append(current_date) highs.append(high) lows.append(low) # 根据最高温度和最低温度绘制图形plt.style.use('seaborn')fig, ax = plt.subplots()ax.plot(dates, highs, c='red', alpha=0.5) # alpha指定颜色的透明度,0为完全透明ax.plot(dates, lows, c='blue', alpha=0.5)ax.fill_between(dates, highs, lows, facecolor='blue',alpha=0.1) # 设置图形的格式ax.set_title("2018年每日最高温度", fontproperties="SimHei", fontsize=24)ax.set_xlabel('', fontproperties="SimHei", fontsize=16)fig.autofmt_xdate()ax.set_ylabel("温度(F)", fontproperties="SimHei", fontsize=16)ax.tick_params(axis='both', which='major', labelsize=16)plt.show()
运行结果如下:
此外,读取CSV文件过程中,数据可能缺失,程序运行时就会报错甚至崩溃。所有需要在从CSV文件中读取值时执行错误检查代码,对可能的异常进行处理,更换数据文件为:death_valley_2018_simple.csv ,该文件有缺失值。
对代码进行修改:
death_valley_highs_lows.py
import csv # 导入csv模块from datetime import datetimeimport matplotlib.pyplot as pltfilename = 'data/death_valley_2018_simple.csv'with open(filename) as f: reader = csv.reader(f) header_row = next(reader) # 返回文件的下一行,在这便是首行,即文件头 # for index, column_header in enumerate(header_row): # 对列表调用了 enumerate()来获取每个元素的索引及其值,方便我们提取需要的数据列 # print(index, column_header) # 从文件中获取日期、最高温度和最低温度 dates, highs, lows = [], [], [] for row in reader: current_date = datetime.strptime(row[2], '%Y-%m-%d') try: high = int(row[5]) low = int(row[6]) except ValueError: print(f"Missing data for {current_date}") else: dates.append(current_date) highs.append(high) lows.append(low) # 根据最高温度和最低温度绘制图形plt.style.use('seaborn')fig, ax = plt.subplots()ax.plot(dates, highs, c='red', alpha=0.5) # alpha指定颜色的透明度,0为完全透明ax.plot(dates, lows, c='blue', alpha=0.5)ax.fill_between(dates, highs, lows, facecolor='blue',alpha=0.1)# 设置图形的格式ax.set_title("2018年每日最高温度和最低气温\n美国加利福利亚死亡谷", fontproperties="SimHei", fontsize=24)ax.set_xlabel('', fontproperties="SimHei", fontsize=16)fig.autofmt_xdate()ax.set_ylabel("温度(F)", fontproperties="SimHei", fontsize=16)ax.tick_params(axis='both', which='major', labelsize=16)plt.show()
如果现在运行 death_valley_highs_lows.py,将会发现缺失数据的日期只有一个:
Missing data for 2018-02-18 00:00:00
妥善地处理错误后,代码能够生成图形并忽略缺失数据的那天。运行结果如下:
感谢各位的阅读,以上就是“Python如何读取CSV文件并进行数据可视化绘图”的内容了,经过本文的学习后,相信大家对Python如何读取CSV文件并进行数据可视化绘图这一问题有了更深刻的体会,具体使用情况还需要大家实践验证。这里是编程网,小编将为大家推送更多相关知识点的文章,欢迎关注!