diff options
Diffstat (limited to '')
| -rw-r--r-- | Tools/PostProcessing/plot_distribution_mapping.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/Tools/PostProcessing/plot_distribution_mapping.py b/Tools/PostProcessing/plot_distribution_mapping.py index 9b229c601..0e732b02d 100644 --- a/Tools/PostProcessing/plot_distribution_mapping.py +++ b/Tools/PostProcessing/plot_distribution_mapping.py @@ -54,10 +54,12 @@ class SimData: self.data_fields, self.keys = data_fields, list(prange) data = np.genfromtxt(directory) + if len(data.shape) == 1: + data = data.reshape(-1, data.shape[0]) + steps = data[:,0].astype(int) times = data[:,1] - reduced_diags_interval = steps[1] - steps[0] data = data[:,2:] # Compute the number of datafields saved per box @@ -103,7 +105,7 @@ class SimData: is_3D = i_is_int and j_is_int and k_is_int for key in self.keys: - row = key//reduced_diags_interval + row = np.where(key == steps)[0][0] costs = data[row, 0::n_data_fields].astype(float) ranks = data[row, 1::n_data_fields].astype(int) icoords = i.astype(int)//i_blocking_factor |