add help remove the exclude list indice from profile list indice

anastasiatarasenko · Apr 1, 2021 · c72b5ff · c72b5ff
1 parent 0a9ac31
commit c72b5ff
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/plots.py b/plots.py
@@ -300,6 +300,8 @@ def section(self, start, end, xaxis, yscale, exclude,
             except:
                 sys.exit("invalid --list {}-{}, max value must be <= {}".format(start, end,
                                                                             profiles[-1]))
+        # remove the exclude list indice from profile list indice
+        # https://www.geeksforgeeks.org/python-indices-list-of-matching-element-from-other-list/?ref=rp
         res = [i for i, val in enumerate(list_profiles) if val in list_exclude]
         list_profiles = np.delete(list_profiles, res)
         #print(list_profiles)