2dim_plot Namespace Reference

Functions
def	cm2inch (value)
def	round_to_divisable (value, divisable)
def	plot_measure2d (xs, ys, zs, xlabel, ylabel, zlabel="", zmin=None, zmax=None, qualitative=False, contour=True, title=None)

Variables
	parser = argparse.ArgumentParser(description='Plot two-dimensional images')
	type
	float
	help
	int
	default
	action
	required
	True
	metavar
	nargs
	args = parser.parse_args()
	simulator = target_file.split('_')[-1].split('.csv')[0]
	experiment = target_file.split('/')[-1].split(simulator)[0]
	results = np.genfromtxt(target_file, delimiter=',', names=True)
	keys = results.dtype.names
	data = np.zeros((results.shape[0], len(keys)))
	fig
	format
	bbox_inches

Function Documentation

def 2dim_plot.cm2inch

(

value

)

Definition at line 54 of file 2dim_plot.py.

def cm2inch(value):
    return value / 2.54

def 2dim_plot.plot_measure2d	(		xs,
			ys,
			zs,
			xlabel,
			ylabel,
			zlabel = "",
			zmin = None,
			zmax = None,
			qualitative = False,
			contour = True,
			title = None
	)

Definition at line 75 of file 2dim_plot.py.

                   zmax=None, qualitative=False, contour=True, title=None):
    fig = plt.figure(figsize=(cm2inch(5.5), cm2inch(5.5)))
    
    ax1 = fig.add_axes([0.0, 0.25, 1.0, 0.85])
    if title is not None:
        plt.title(title)
    ax2 = fig.add_axes([0.0, 0.0, 1.0, 0.05])
    

    _, steps_x = np.unique(xs, return_counts=True)
    _, steps_y = np.unique(ys, return_counts=True)
    steps_x = np.max(steps_x)
    steps_y = np.max(steps_y)
    xs = xs.reshape((steps_y, steps_x))
    ys = ys.reshape((steps_y, steps_x))
    zs = zs.reshape((steps_y, steps_x))
    zs = zs.transpose()

    # Auto-scale
    idcs = np.isfinite(zs)
    if np.sum(idcs) == 0:
        return
    if zmin is None:
        zmin = np.min(zs[idcs])
        if 0 < zmin < 1:
            zmin = 0
        else:
            zmin = int(zmin)
    if zmax is None:
        zmax = round_to_divisable(np.max(zs[idcs]), args.nl - 1)
        if zmin > 0:
            zmax = zmax + zmin
        if 0 < zmax < 1:
            zmax = 1

    # Select the colormap
    if qualitative:
        cmap = plt.cm.rainbow
    else:
        #cmap = plt.cm.Purples
        # if zmin < 0.0:
        cmap = plt.cm.PuOr
    cmap.set_bad('black', 1.)

    extent = (np.min(xs), np.max(xs), np.min(ys), np.max(ys))
    ax1.imshow(zs, aspect='auto', origin='lower', extent=extent, cmap=cmap,
               vmin=zmin, vmax=zmax, interpolation="none")

    levels = np.linspace(zmin, zmax, args.nl)
    zs = zs.transpose()
    if contour:
        CS2 = ax1.contour(xs, ys, zs, levels, linewidths=0.25,
                          colors='k', vmin=zmin, vmax=zmax)
    ax1.grid(color='black', linestyle=':', linewidth=0.25)
    ax1.set_xlabel(xlabel)
    ax1.set_ylabel(ylabel)
    cbar = matplotlib.colorbar.ColorbarBase(ax2, cmap=cmap,
                                            orientation='horizontal', ticks=levels,
                                            norm=matplotlib.colors.Normalize(zmin, zmax))
    cbar.set_label(zlabel)
    return fig

def 2dim_plot.round_to_divisable	(		value,
			divisable
	)

Definition at line 58 of file 2dim_plot.py.

def round_to_divisable(value, divisable):
    if value == 0:
        return 0
    temp = np.abs(value)
    a = 0
    while temp < divisable:
        temp *= 10.0
        a += 1
    if temp % divisable == 0:
        return value
    res = old_div((temp - (temp % divisable) + divisable), (10.0**a))
    if value < 0:
        return -res
    return res

Variable Documentation

2dim_plot.action

Definition at line 36 of file 2dim_plot.py.

2dim_plot.args = parser.parse_args()

Definition at line 44 of file 2dim_plot.py.

2dim_plot.bbox_inches

Definition at line 163 of file 2dim_plot.py.

2dim_plot.data = np.zeros((results.shape[0], len(keys)))

Definition at line 148 of file 2dim_plot.py.

2dim_plot.default

Definition at line 35 of file 2dim_plot.py.

2dim_plot.experiment = target_file.split('/')[-1].split(simulator)[0]

Definition at line 143 of file 2dim_plot.py.

2dim_plot.fig

Initial value:

= plot_measure2d(data[:, 0], data[:, 1], data[:, args.z],
                         xlabel=get_label(keys[0]), ylabel=get_label(keys[1]),
                         zlabel=get_label(keys[args.z]), zmin=args.zmin,
                         zmax=args.zmax, qualitative=args.q, contour=args.c,
                         title=SIMULATOR_LABELS[simulator])

Definition at line 152 of file 2dim_plot.py.

2dim_plot.float

Definition at line 32 of file 2dim_plot.py.

2dim_plot.format

Definition at line 162 of file 2dim_plot.py.

2dim_plot.help

Definition at line 32 of file 2dim_plot.py.

2dim_plot.int

Definition at line 35 of file 2dim_plot.py.

2dim_plot.keys = results.dtype.names

Definition at line 147 of file 2dim_plot.py.

2dim_plot.metavar

Definition at line 42 of file 2dim_plot.py.

2dim_plot.nargs

Definition at line 42 of file 2dim_plot.py.

2dim_plot.parser = argparse.ArgumentParser(description='Plot two-dimensional images')

Definition at line 29 of file 2dim_plot.py.

2dim_plot.required

Definition at line 41 of file 2dim_plot.py.

2dim_plot.results = np.genfromtxt(target_file, delimiter=',', names=True)

Definition at line 146 of file 2dim_plot.py.

2dim_plot.simulator = target_file.split('_')[-1].split('.csv')[0]

Definition at line 142 of file 2dim_plot.py.

2dim_plot.True

Definition at line 41 of file 2dim_plot.py.

2dim_plot.type

Definition at line 32 of file 2dim_plot.py.