plot_heatmaps.py 文件源码

def plot_func(cause, effect, seasonal, different, heatmap_func, title):
    heatmap = numpy.zeros((len(sigma01_vals), len(sd_proc_vals)))

    eps = eps_vals[seasonal]
    beta00 = beta00_vals[different]

    for i, sigma01 in enumerate(sigma01_vals):
        for j, sd_proc in enumerate(sd_proc_vals):
            heatmap[i,j] = heatmap_func(cause, effect, eps, beta00, sigma01, sd_proc)
    print heatmap

    plot_heatmap(heatmap,
        'sd_proc', ['{0:.2g}'.format(x) for x in sd_proc_vals],
        'sigma01', ['{0:.2g}'.format(y) for y in sigma01_vals],
        vmin=0, vmax=1
    )
    pyplot.title('{}: {} causes {}'.format(title, cause, effect))