aboutsummaryrefslogtreecommitdiffstats
path: root/python/analytics/utils.py
blob: c499c65f035d06553665224483c7810e2d13b852 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
import numpy as np
from scipy.special import h_roots
from dateutil.relativedelta import relativedelta
import datetime

def GHquad(n):
    """Gauss-Hermite quadrature weights"""
    Z, w = h_roots(n)
    return Z*np.sqrt(2), w/np.sqrt(np.pi)

def previous_twentieth(d):
    r = datetime.date(d.year, d.month, 20)
    if r > d:
        r -= relativedelta(months=1)
    mod = r.month % 3
    if mod != 0:
        r -= relativedelta(months = mod)
    return r