pyvibdmc.analysis.plotter

Classes

Plotter

A very basic plotting class that will use matplotlib to generate various plots using results from PyVibDMC/AnalyzeWfn/SimInfo.

Module Contents

class pyvibdmc.analysis.plotter.Plotter[source]

A very basic plotting class that will use matplotlib to generate various plots using results from PyVibDMC/AnalyzeWfn/SimInfo.

params
static plt_vref_vs_tau(vref_vs_tau, save_name='vref_vs_tau.png')[source]

Takes in the vref vs tau array from a DMC sim and plots it. Can also take in many vref_vs_tau arrays and plot them successively

Parameters:

  • vref_vs_tau (str or list) – The vref_vs_tau array from the *sim_info.hdf5 file. Can be a list of these as well.

  • save_name (str) – The name of the plot that will save as a .png

Returns:

static plt_pop_vs_tau(pop_vs_tau, save_name='pop_vs_tau.png')[source]

Takes in the pop_vs_tau array from a DMC sim and plots it. Can also take in many pop_vs_tau arrays and plot them successively

Parameters:

  • pop_vs_tau (str or list) – The vref_vs_tau array from the *sim_info.hdf5 file. Can be a list of these as well.

  • save_name (str) – The name of the plot that will save as a .png

Returns:

static plt_hist1d(hist, xlabel, ylabel='Probability Amplitude ($\\mathrm{\\Psi^{2}}$)', title='', save_name='histogram.png')[source]

Plots the histogram generated from AnalyzeWfn.projection_1d.

Parameters:

  • hist (np.ndarray) – Output from AnalyzeWfn.projection_1d ; array of shape (bin_num-1 x 2)

  • xlabel (str) – What to label the x-axis

  • ylabel (str) – What to label the y-axis (units in Parenthesis is always good)

  • title (str) – Title of the plot

  • save_name (str) – name of the .png file that this plot will be saved to

static plt_hist2d(binsx, binsy, hist_2d, xlabel, ylabel, title='', save_name='histogram.png')[source]

Plots the 2D-histogram generated from AnalyzeWfn.projection_2d.

Parameters:

  • bins (np.ndarray) – First output from AnalyzeWfn.projection_2d; the array containing the x and y bins

  • hist_2d (np.ndarray) – Second output from AnalyzeWfn.projection_2d; The matrix that contains the amplitude at each of the bins in the 2d histogram.

  • xlabel (str) – What to label the x-axis

  • ylabel (str) – What to label the y-axis (units in Parenthesis is always good)

  • title (str) – Title of the plot

  • save_name (str) – name of the .png file that this plot will be saved to