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

Make colour gradient work as intended for subsets

Browse source
This commit is contained in:
rasmusvt 2022-08-18 09:39:14 +02:00
parent 8ca73e4687
commit 9f2199e69a
1 changed files with 6 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

9
nafuma/xanes/plot.py
View file

@ -38,16 +38,19 @@ def plot_xanes(data, options={}):
# Update list of cycles to correct indices # Update list of cycles to correct indices
update_scans_list(data=data, options=options) update_scans_list(data=data, options=options)
colours = generate_colours(scans=data['path'], options=options) colours = generate_colours(scans=options['which_scans'], options=options)
# Prepare plot, and read and process data # Prepare plot, and read and process data
fig, ax = btp.prepare_plot(options=options) fig, ax = btp.prepare_plot(options=options)
# Add counter to pick out correct colour
counter = 0
for i, path in enumerate(data['path']): for i, path in enumerate(data['path']):
if i in options['which_scans']: if i in options['which_scans']:
data['xanes_data'].plot(x='ZapEnergy', y=path, ax=ax, c=colours[i]) data['xanes_data'].plot(x='ZapEnergy', y=path, ax=ax, c=colours[counter])
counter += 1
fig, ax = btp.adjust_plot(fig=fig, ax=ax, options=options) fig, ax = btp.adjust_plot(fig=fig, ax=ax, options=options)