Python script to construct a snow depth figure

makeSnowdepthFig.py

"""
Snow depth figure
version 6 April 2018
by Antti Lipponen

Copyright (c) 2018 Antti Lipponen

Permission is hereby granted, free of charge, to any person obtaining
a copy of this software and associated documentation files (the
"Software"), to deal in the Software without restriction, including
without limitation the rights to use, copy, modify, merge, publish,
distribute, sublicense, and/or sell copies of the Software, and to
permit persons to whom the Software is furnished to do so, subject to
the following conditions:
The above copyright notice and this permission notice shall be included
in all copies or substantial portions of the Software.
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
"""

# imports
import numpy as np
import pandas as pd
from seaborn import cubehelix_palette
import matplotlib.pyplot as plt
import matplotlib as mpl
# from imageio import imread  # uncomment if you have a "background image"
from datetime import datetime

# default font: Lato
mpl.rc('font', family='Lato')

# Data I have used here is the Finnish Meteorological Institute snow depth open data
# downloaded from: http://ilmatieteenlaitos.fi/havaintojen-lataus/
# English web site: http://en.ilmatieteenlaitos.fi/download-observations

# options & title texts
case = 'Helsinki'  # in this case the data should be in snowHelsinki.csv
stationFI = 'Kaisaniemi, Helsinki'  # otsikkoteksti suomeksi
stationEN = 'Kaisaniemi, Helsinki'  # otsikkoteksti englanniksi

#########################

# If you have a "background image" for the figure
# bgImg = imread('bg.png')  # a png image of size 1920 x 1080

# colormap from Seaborn cubehelix palette
cmap = cubehelix_palette(n_colors=19, start=2.85, rot=.1, as_cmap=True, reverse=False)
norm = mpl.colors.Normalize(vmin=0, vmax=19)

# get data
s = pd.read_csv('snow{}.csv'.format(case))

# snow depth (all days, limit to 0 - 9999 cm)
snowDepth = np.clip(np.array(s['Lumensyvyys (cm)']), a_min=0, a_max=9999)  # if you use English data change (Lumensyvyys = snow depth)

# load dates (there are better ways to do this but this works as well)
vv, kk, pp = np.array(s['Vuosi']), np.array(s['Kk']), np.array(s['Pv'])
dates = []
for ii in range(len(vv)):
    dates.append(datetime(int(vv[ii]), int(kk[ii]), int(pp[ii])))

# Create figure
fig = plt.figure(figsize=(19.2, 10.8))

# Axis for background
axBG = fig.add_axes([0.0, 0.0, 1.0, 1.0])
axBG.axis('off')
# axBG.imshow(bgImg, origin='upper', aspect='auto', extent=[0, 1, 0, 1])  # uncomment if you have a background image
axBG.set_xlim([0, 1])
axBG.set_ylim([0, 1])
# title texts
axBG.text(0.006 + 0.0015, 0.987 - 0.0015, 'Lumensyvyys – {}'.format(stationFI),
          fontsize=48, color='#4D4D4D', horizontalalignment='left',
          verticalalignment='top', clip_on=False)
axBG.text(0.006, 0.987, 'Lumensyvyys – {}'.format(stationFI), fontsize=48,
          color='#E5E5E5', horizontalalignment='left', verticalalignment='top', clip_on=False)
axBG.text(0.006 + 0.0015, 0.92 - 0.0015, 'Snow depth – {}, Finland'.format(stationEN),
          fontsize=38, color='#4D4D4D', horizontalalignment='left',
          verticalalignment='top', clip_on=False)
axBG.text(0.006, 0.92, 'Snow depth – {}, Finland'.format(stationEN), fontsize=38,
          color='#E5E5E5', horizontalalignment='left', verticalalignment='top', clip_on=False)

# names of months
monthsFIN = ['Tam', 'Hel', 'Maa', 'Huh', 'Tou', 'Kes', 'Hei', 'Elo', 'Syy', 'Lok', 'Mar', 'Jou']
monthsENG = ['Jan', 'Feb', 'Mar', 'Apr', 'May', 'Jun', 'Jul', 'Aug', 'Sep', 'Oct', 'Nov', 'Dec']

# axis for months
axTimes = fig.add_axes([0.1, 0.1, 0.85, 0.8])
axTimes.axis('off')
axTimes.set_xlim([90, 340])  # show days 90-340 (day 0 = July 1st)
axTimes.set_ylim([0, 1])

# axis according to winter 2017 - 2018
N = (datetime(2018, 6, 1) - datetime(2017, 7, 1)).days  # days to 1st of June
axTimes.plot([N, N], [0.05, 0.9], c='#BBBBBB', linewidth=0.5, linestyle='-', zorder=ii, clip_on=False)
for ii in [10, 11, 12, 1, 2, 3, 4, 5]:
    if ii < 7:
        N = (datetime(2018, ii, 1) - datetime(2017, 7, 1)).days
    else:
        N = (datetime(2017, ii, 1) - datetime(2017, 7, 1)).days
    axTimes.plot([N, N], [0.05, 0.9], c='#BBBBBB', linewidth=0.5, linestyle='-', clip_on=False, zorder=1)
    # write the name of the month at 15. day
    axTimes.text(N + 15, 0.035, monthsFIN[ii - 1], fontsize=16, color='#DEDEDE', horizontalalignment='center', verticalalignment='top', clip_on=False, zorder=1)
    axTimes.text(N + 15, 0.005, monthsENG[ii - 1], fontsize=16, color='#DEDEDE', horizontalalignment='center', verticalalignment='top', clip_on=False, zorder=1)

# axis for the scale
axScale = fig.add_axes([0.1, 0.85, 0.85, 0.125])
axScale.axis('off')
axScale.set_xlim([90, 340])
axScale.set_ylim([0, 100])
axScale.plot([322, 322], [0, 100], linestyle='-', c='#dedede', linewidth=1.5, clip_on=False)
axScale.plot([322 - 1.75, 322 + 1.75], [0, 0], linestyle='-', c='#dedede', linewidth=1.5, clip_on=False)
axScale.plot([322 - 1.75, 322 + 1.75], [100, 100], linestyle='-', c='#dedede', linewidth=1.5, clip_on=False)
axScale.text(322 + 5, 50, '100 cm', fontsize=17, color='#DEDEDE', horizontalalignment='left', verticalalignment='center', clip_on=False)
for ii in range(1, 10):
    axScale.plot([322 - 0.35, 322 + 0.35], [ii * 10, ii * 10], linestyle='-', c='#dedede', linewidth=1.5, clip_on=False)

# draw snow depths
for ii, yy in enumerate(range(2006, 2017 + 1)):
    axSnow = fig.add_axes([0.1, 0.78 - 0.058 * ii, 0.85, 0.125])
    axSnow.axis('off')
    axSnow.set_xlim([90, 340])  # days 90 - 340
    axSnow.set_ylim([0, 100])  # 0 - 100 cm
    axSnow.plot([90, 340], [0, 0], c='#DEDEDE', linewidth=1.5, linestyle='-', zorder=100 + ii, clip_on=False)  # draw level 0 cm
    axSnow.text(85, 10, '{} – {}'.format(yy, yy + 1), fontsize=16,
                color='#DEDEDE', horizontalalignment='right',
                verticalalignment='center', clip_on=False)  # write years

    # days from 1st July, then filter out days outside range 90-340
    dd = np.array([(xx - datetime(yy, 7, 1)).days for xx in dates])
    MASK = np.logical_and(dd >= 90, dd <= 340)
    axSnow.fill_between(dd[MASK], snowDepth[MASK], color=cmap(norm(2 + ii)), alpha=1.0, zorder=100 + ii, clip_on=False)
    axSnow.plot(dd[MASK], snowDepth[MASK], color='#F3F3F3', linewidth=2.5, linestyle='-',
                alpha=1.0, zorder=100 + ii * 10, clip_on=False)

# save image
plt.savefig('snowdepth_{}.png'.format(case), dpi=100, transparent=True)
# close figure
plt.close()

snowHelsinki.csv