You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
115 lines
2.8 KiB
115 lines
2.8 KiB
# -*- coding: utf-8 -*-
|
|
|
|
import csv
|
|
from dataclasses import dataclass
|
|
from datetime import datetime
|
|
|
|
import cartopy.crs as ccrs
|
|
import matplotlib.pyplot as plt
|
|
import cartopy.feature as cfeature
|
|
from mpl_toolkits.axes_grid1.inset_locator import inset_axes
|
|
from cartopy import geodesic
|
|
import shapely.geometry as sgeom
|
|
|
|
|
|
proj = ccrs.AlbersEqualArea(
|
|
central_longitude=-95,
|
|
central_latitude=40,
|
|
)
|
|
water_blue = "#7ebfd4"
|
|
|
|
|
|
def phys(name, resolv):
|
|
return cfeature.NaturalEarthFeature('physical', name, resolv, facecolor="none")
|
|
|
|
|
|
def cultural(name, resolv):
|
|
return cfeature.NaturalEarthFeature('cultural', name, resolv, facecolor="none")
|
|
|
|
|
|
land = phys("land", "50m")
|
|
rivers = phys("rivers_lake_centerlines", "50m")
|
|
lakes = phys("lakes", "50m")
|
|
countries = cultural("admin_0_countries", "50m")
|
|
states = cultural("admin_1_states_provinces_lines", "50m")
|
|
|
|
|
|
@dataclass
|
|
class Tornado:
|
|
om: str
|
|
yr: str
|
|
mo: str
|
|
dy: str
|
|
date: str
|
|
time: str
|
|
tz: str
|
|
st: str
|
|
stf: str
|
|
stn: str
|
|
mag: str
|
|
inj: str
|
|
fat: str
|
|
loss: str
|
|
closs: str
|
|
slat: str
|
|
slon: str
|
|
elat: str
|
|
elon: str
|
|
len: str
|
|
wid: str
|
|
ns: str
|
|
sn: str
|
|
sg: str
|
|
f1: str
|
|
f2: str
|
|
f3: str
|
|
f4: str
|
|
fc: str
|
|
|
|
def __post_init__(self):
|
|
self.id = int(self.om)
|
|
self.dt = datetime.strptime(self.date, "%Y-%m-%d")
|
|
self.slon = float(self.slon)
|
|
self.slat = float(self.slat)
|
|
self.elon = float(self.elon)
|
|
self.elat = float(self.elat)
|
|
|
|
def __repr__(self):
|
|
return "<tornado {id} {date}>".format(id=self.id, date=self.dt)
|
|
|
|
|
|
def read_data():
|
|
tornados = []
|
|
with open("1950-2017_actual_tornadoes.csv", newline='') as csvfile:
|
|
reader = csv.DictReader(csvfile)
|
|
for i, row in enumerate(reader):
|
|
tornados.append(Tornado(**row))
|
|
return tornados
|
|
|
|
|
|
def draw_map(data):
|
|
fig = plt.figure(frameon=False)
|
|
ax = fig.add_subplot(111, projection=proj)
|
|
ax.background_patch.set_facecolor(water_blue)
|
|
ax.set_extent([-125, -65, 20, 50], ccrs.Geodetic())
|
|
ax.add_feature(land, facecolor="#f0f0f0")
|
|
ax.add_feature(rivers, edgecolor=water_blue)
|
|
ax.add_feature(lakes, facecolor=water_blue)
|
|
|
|
ax.add_feature(countries, edgecolor="grey", linewidth=0.2, alpha=0.6, dashes='--')
|
|
ax.add_feature(states, edgecolor="grey", linewidth=0.2, alpha=0.4, dashes='--')
|
|
|
|
for i, t in enumerate(data):
|
|
print(i)
|
|
if t.elon < 0:
|
|
ax.plot([t.slon, t.elon], [t.slat, t.elat], 'r', lw=0.3, transform=ccrs.Geodetic())
|
|
else:
|
|
ax.plot([t.slon, t.slon], [t.slat, t.slat], 'r', lw=0.3, transform=ccrs.Geodetic())
|
|
|
|
fig.tight_layout()
|
|
plt.savefig("tornados_us.png", dpi=320)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
ts = read_data()
|
|
draw_map(ts)
|