Python/ke_lab8_graph.py at master · kodyellis/Python · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
#Kody Ellis
#Lab8 Plot
import numpy as np

def WeatherStats(aWeatherFile):
    barometric_p = []
    relative_humidity = []
    temperature = []
    wind_direction = []
    wind_speed = []

    fileHandle = open(aWeatherFile, "r")
    for item in fileHandle:
        if (item[0] == "#") or (not(item[0].isdigit())):
            pass
        else:
            second = item.split()[1]
            if second.split(',')[1] == 'NaN':
                pass
            else:
                barometric_p.append(float(second.split(',')[1]))

            if second.split(',')[2] == 'NaN':
                pass
            else:
                relative_humidity.append(float(second.split(',')[2]))

            if second.split(',')[3] == 'NaN':
                pass
            else:
                temperature.append(float(second.split(',')[3]))

            if second.split(',')[4] == 'NaN':
                pass
            else:
                wind_direction.append(float(second.split(',')[4]))

            if second.split(',')[5] == 'NaN':
                pass
            else:
                wind_speed.append(float(second.split(',')[5]))

    if len(barometric_p) ==  0:
        fileHandle.close()
        return None
    else:
        out5 = sum(wind_speed)/len(wind_speed)

        fileHandle.close()
        return(out5)


import glob
file_list = glob.glob('Augspurger/*')

monthly_bp_average = []
fall_season_monthly_bp_average = []
winter_season_monthly_bp_average = []
spring_season_monthly_bp_average = []
summer_season_monthly_bp_average = []

#tupless
fall = ("9","10","11")
winter = ("12","01","02")
spring = ("03","04","05")
summer = ("06","07","08")

fall_string = 'Monthly Wind Speed Average Sep to Nov 2016 - 2017'
winter_string = 'Monthly Wind Speed Average Dec to Feb 2016 - 2017'
spring_string = 'Monthly Wind Speed Average Mar to May 2016 - 2017'
summer_string = 'Monthly Wind Speed Average Jun to Aug 2016 - 2017'

def season_to_list(list1,season):
    for individual_file in file_list:
        flag = individual_file.split('_')[2].split('_')[0].split('.')[0] #finally gets 01, 02, etc
        if flag in season:
            out = WeatherStats(individual_file)
            list1.append(out)


season_to_list(fall_season_monthly_bp_average, fall)
season_to_list(winter_season_monthly_bp_average, winter)
season_to_list(spring_season_monthly_bp_average, spring)
season_to_list(summer_season_monthly_bp_average, summer)

fall_avg = np.mean(fall_season_monthly_bp_average)
winter_avg = np.mean(winter_season_monthly_bp_average)
spring_avg = np.mean(spring_season_monthly_bp_average)
summer_avg = np.mean(summer_season_monthly_bp_average)

month_avg = [fall_avg, winter_avg,spring_avg,summer_avg]

import matplotlib.pyplot as plt

def graph(season_list, word):
    plt.plot(season_list)
    plt.xlabel('Indices of Months', fontsize = 15)
    plt.ylabel('Wind Speed Average',fontsize = 15)
    plt.title(word, fontsize = 14)
    plt.savefig('average_speed_seasons.png')
    plt.show()

graph(fall_season_monthly_bp_average, fall_string)
graph(winter_season_monthly_bp_average, winter_string)
graph(spring_season_monthly_bp_average, spring_string)
graph(summer_season_monthly_bp_average, summer_string)