In [1]:
from monthly_totals import *
In [2]:
data = get_precip_data()
In [3]:
totals_pd, counts_pd = pivot_months_pandas(data)
In [4]:
totals_loop, counts_loop = pivot_months_loops(data)
In [5]:
totals_loop # look how nicely Jupyter displays a DataFrame...
Out[5]:
In [6]:
abs(totals_pd - totals_loop).values.sum() # the two totals should be the same
Out[6]:
In [7]:
abs(counts_pd - counts_loop).values.sum() # ...counts too.
Out[7]:
In [8]:
%timeit pivot_months_pandas(data)
In [9]:
%timeit pivot_months_loops(data)