rasmus/nafuma
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Change default behaviour with 10+ plots

Browse source
This commit is contained in:
rasmusvt 2022-03-30 17:40:06 +02:00
parent 0fbfd20a74
commit 223be18c3e
3 changed files with 24 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

20
beamtime/plotting.py
View file

@ -377,16 +377,20 @@ def update_rc_params(rc_params):
plt.rcParams.update({key: rc_params[key]})
def generate_colours(palettes):
def generate_colours(palettes, kind=None):
# Creates a list of all the colours that is passed in the colour_cycles argument. Then makes cyclic iterables of these.
colour_collection = []
for palette in palettes:
mod = importlib.import_module("palettable.colorbrewer.%s" % palette[0])
colour = getattr(mod, palette[1]).mpl_colors
colour_collection = colour_collection + colour
if kind == 'single':
colour_cycle = itertools.cycle(palettes)
colour_cycle = itertools.cycle(colour_collection)
else:
# Creates a list of all the colours that is passed in the colour_cycles argument. Then makes cyclic iterables of these.
colour_collection = []
for palette in palettes:
mod = importlib.import_module("palettable.colorbrewer.%s" % palette[0])
colour = getattr(mod, palette[1]).mpl_colors
colour_collection = colour_collection + colour
colour_cycle = itertools.cycle(colour_collection)
return colour_cycle