Commit 41eb5183 authored by syz's avatar syz
Browse files

plot_complex_loop_stack now corrects num_comps via min

parent d2f7b702
...@@ -629,6 +629,8 @@ def plot_complex_map_stack(map_stack, num_comps=4, title=None, x_label='', y_lab ...@@ -629,6 +629,8 @@ def plot_complex_map_stack(map_stack, num_comps=4, title=None, x_label='', y_lab
figsize = kwargs.pop('figsize', (4, 4)) figsize = kwargs.pop('figsize', (4, 4))
figsize = (figsize[0] * num_comps, 8) figsize = (figsize[0] * num_comps, 8)
num_comps = min(num_comps, map_stack.shape[0])
fig, axes = plt.subplots(2, num_comps, figsize=figsize) fig, axes = plt.subplots(2, num_comps, figsize=figsize)
fig.subplots_adjust(hspace=0.1, wspace=0.4) fig.subplots_adjust(hspace=0.1, wspace=0.4)
if title is not None: if title is not None:
...@@ -687,6 +689,8 @@ def plot_complex_loop_stack(loop_stack, x_vec, title=None, subtitle_prefix='Comp ...@@ -687,6 +689,8 @@ def plot_complex_loop_stack(loop_stack, x_vec, title=None, subtitle_prefix='Comp
if min(num_comps, loop_stack.shape[0]) == 1: if min(num_comps, loop_stack.shape[0]) == 1:
subtitle_prefix = None subtitle_prefix = None
num_comps = min(num_comps, loop_stack.shape[0])
funcs = [np.abs, np.angle] funcs = [np.abs, np.angle]
comp_labs = ['Amplitude (' + amp_units + ')', 'Phase (rad)'] comp_labs = ['Amplitude (' + amp_units + ')', 'Phase (rad)']
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment