aboutsummaryrefslogtreecommitdiffstats
path: root/python/analytics/utils.py
blob: ddfc6b23d6a2572809ca0992e8fa2d77840c1377 (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 : int):
    """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