Skip to content

Commit

Permalink
Some plotting aestetics
Browse files Browse the repository at this point in the history
  • Loading branch information
hephysicist committed Jan 20, 2025
1 parent 188f83e commit b0dc5f6
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions columnflow/plotting/plot_functions_1d.py
Original file line number Diff line number Diff line change
Expand Up @@ -71,7 +71,8 @@ def plot_variable_per_process(
total_events = {key: sum(hist.values()) for key, hist in hists.items()}

# Sort processes by total number of events in descending order
sorted_hists_desc = OrderedDict(sorted(hists.items(), key=lambda item: total_events[item[0]], reverse=True))
#sorted_hists_desc = OrderedDict(sorted(hists.items(), key=lambda item: total_events[item[0]], reverse=True))
sorted_hists_desc = OrderedDict(hists.items())

# Get keys of sorted processes
sorted_keys = list(sorted_hists_desc.keys())
Expand All @@ -85,7 +86,7 @@ def plot_variable_per_process(
custom_order = sorted_keys
else:
# More than two processes, custom order: highest, rest, then second highest
custom_order = [sorted_keys[0]] + sorted_keys[2:] + [sorted_keys[1]]
custom_order = sorted_keys #[sorted_keys[0]] + sorted_keys[2:] + [sorted_keys[1]]

# Reorder histograms based on custom order
sorted_hists = OrderedDict((key, sorted_hists_desc[key]) for key in custom_order)
Expand Down

0 comments on commit b0dc5f6

Please sign in to comment.