Commit 00b822a5 authored by Tom Rodemund's avatar Tom Rodemund 🦆
Browse files

fixed small bug

parent 97d0ba80
......@@ -824,9 +824,9 @@ def main():
# define columns, columns correspond to the position in q
# s in [0,1] -> cavity 1, s in [1,2] -> cavity 2, and so on
comparison_q_list = np.concatenate([q_uselist + ii for ii in range(N_CAVITIES)])
q_list_complete = np.concatenate([q_uselist + ii for ii in range(N_CAVITIES)])
# first column is list of impulses, thus p
df_columns = ['p'] + list(comparison_q_list)
df_columns = ['p'] + list(q_list_complete)
husimis_inc = np.concatenate([
[husimi_inc_list[jj*N_CAVITIES+ii] for jj in range(n_qpoints)]
......@@ -877,7 +877,7 @@ def main():
):
plot_husimi(
q_uselist, impulses_henceforth, husimi_use.T,
q_list_complete, impulses_henceforth, husimi_use.T,
inout_index=inout_index,
wave_direction=wavedirection,
savedir=timestepdir_now,
......
Supports Markdown
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