-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathutils.py
175 lines (147 loc) · 6.12 KB
/
utils.py
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
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
import argparse
from geopy.point import Point
from geopy.units import kilometers
import geopy.distance
def load_state_abbrevs():
"""Utility function of each state's abbreviations
Returns:
========
state_abbrevs (dict): A dictionary containing each state's abbreviations
"""
state_abbrevs = {"AL": "Alabama",
"AK": "Alaska",
"AZ": "Arizona",
"AR": "Arkansas",
"CA": "Calfornia",
"CO": "Colorado",
"CT": "Connecticut",
"DE": "Delaware",
"FL": "Florida",
"GA": "Georgia",
"HI": "Hawaii",
"ID": "Idaho",
"IL": "Illinois",
"IN": "Indiana",
"IA": "Iowa",
"KS": "Kansas",
"KY": "Kentucky",
"LA": "Louisiana",
"ME": "Maine",
"MD": "Maryland",
"MA": "Massachusetts",
"MI": "Michigan",
"MN": "Minnesota",
"MS": "Mississippi",
"MO": "Missouri",
"MT": "Montana",
"NE": "Nebraska",
"NV": "Nevada",
"NH": "New Hampshire",
"NJ": "New Jersey",
"NM": "New Mexico",
"NY": "New York",
"NC": "North Carolina",
"ND": "North Dakota",
"OH": "Ohio",
"OK": "Oklahoma",
"OR": "Oregon",
"PA": "Pennsylvania",
"PR": "Puerto Rico",
"RI": "Rhode Island",
"SC": "South Carolina",
"SD": "South Dakota",
"TN": "Tennessee",
"TX": "Texas",
"UT": "Utah",
"VT": "Vermont",
"VA": "Virginia",
"WA": "Washington",
"WV": "West Virginia",
"WI": "Wisconsin",
"WY": "Wyoming"}
return state_abbrevs
def calculate_surrounding_coords(sample, num_miles):
"""The project asks to address the question "How many [statistic] COVID-19 occurred
within [num_miles] from [US County]?"
My approach was to calculate the surrounding coordinates from the base coordinate within
the user specified num_miles radius and zoom in on that area in Plotly.
Args:
=====
sample (pd.DataFrame): The pandas DataFrame filtered by user-specified county
num_miles (int) : The number of miles between 0 and 1000
Returns:
=======
lat (list): A list that contains the latitude values from NSEW
lon (list): A list that contains the longitude values from NSEW
"""
miles_to_km = kilometers(miles = num_miles)
start = Point(sample["latitude"].values[0], sample["longitude"].values[0])
d = geopy.distance.geodesic()
dest_north = d.destination(start, distance = miles_to_km, bearing = 0)
dest_east = d.destination(start, distance = miles_to_km, bearing = 90)
dest_south = d.destination(start, distance = miles_to_km, bearing = 180)
dest_west = d.destination(start, distance = miles_to_km, bearing = 270)
lat = [dest_north.latitude, dest_east.latitude, dest_south.latitude, dest_west.latitude]
lon = [dest_north.longitude, dest_east.longitude, dest_south.longitude, dest_west.longitude]
return lat, lon
def check_county(df, county):
"""Check to make sure user-specified county is in the dataframe
Args:
=====
df (pd.DataFrame): A pandas DataFrame from load_data()
county (str): User-specified county of interest
Raises:
=======
Exception: Error message will show if user-specified county is not in dataframe
Returns:
county (str): User-specified county that is in the dataframe
"""
for counties in df["county"]:
if county in counties:
return county
raise Exception("The inputted county does not have any COVID-19 related information. Please try another county.")
def check_state(df, state):
"""Check to make sure user-specified state is in the dataframe
Args:
=====
df (pd.DataFrame): A pandas DataFrame from load_data()
state (str): User-specified state of interest
Raises:
=======
Exception: Error message will show if user-specified state is not in dataframe
Returns:
state (str): User-specified state that is in the dataframe
"""
for states in df["state"]:
if state in states:
return state
raise Exception("The inputted state does not exist. Please input a valid state.")
def load_args():
"""Utility function to load the command line arguments
"""
parser = argparse.ArgumentParser()
parser.add_argument("--county",
type = str,
default = "Barnstable County, MA",
required = True,
help = """
The county and the state it's in.
Please specify as so: 'Barnstable County, MA'
""")
parser.add_argument("--statistic",
type = str,
default = "cases",
required = True,
help = """
The COVID-19 statistic of interest.
Options are: 'cases', 'deaths', 'confirmed_cases',
'confirmed_cases', 'confirmed_deaths',
'probable_cases', 'probable_deaths'
""")
parser.add_argument("--num_miles",
type = int,
default = 0,
required = True,
help = "A number between 0 and 1000")
args = parser.parse_args()
return args