读取CSV内容并用matplotlib做可视化

admin

读取CSV内容并用matplotlib做可视化：360行、24列的CSV数据，读取其中的360行，和3列（日期、最高温、最低温），然后可视化输出。
源码和文件：

1. import csv
   
2. from datetime import datetime
   
3. 
   
4. from matplotlib import pyplot as plt
   
5. 
   
6. # Get dates, high, and low temperatures from file.
   
7. filename = 'death_valley_2014.csv'
   
8. with open(filename) as f:
   
9.     reader = csv.reader(f)
   
10.     header_row = next(reader)
    
11. 
    
12.     dates, highs, lows = [], [], []
    
13.     for row in reader:
    
14.         try:
    
15.             current_date = datetime.strptime(row[0], "%Y-%m-%d")
    
16.             high = int(row[1])
    
17.             low = int(row[3])
    
18.         except ValueError:
    
19.             print(current_date, 'missing data')
    
20.         else:
    
21.             dates.append(current_date)
    
22.             highs.append(high)
    
23.             lows.append(low)
    
24. 
    
25. # Plot data.
    
26. fig = plt.figure(dpi=128, figsize=(10, 6))
    
27. plt.plot(dates, highs, c='red', alpha=0.5)
    
28. plt.plot(dates, lows, c='blue', alpha=0.5)
    
29. plt.fill_between(dates, highs, lows, facecolor='blue', alpha=0.1)
    
30. 
    
31. # Format plot.
    
32. title = "Daily high and low temperatures - 2014\nDeath Valley, CA"
    
33. plt.title(title, fontsize=20)
    
34. plt.xlabel('', fontsize=16)
    
35. fig.autofmt_xdate()
    
36. plt.ylabel("Temperature (F)", fontsize=16)
    
37. plt.tick_params(axis='both', which='major', labelsize=16)
    
38. 
    
39. plt.show()
    

复制代码

执行结果：