Tweaking plot aesthetics

epfl-theos · Jun 19, 2024 · 645147f · 645147f
1 parent dc7d968
commit 645147f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/tutorials/tutorial_6/cri3_bandstructures.png b/tutorials/tutorial_6/cri3_bandstructures.png
diff --git a/tutorials/tutorial_6/plot_bands.py b/tutorials/tutorial_6/plot_bands.py
@@ -1,6 +1,7 @@
-from koopmans.io import read
 import matplotlib.pyplot as plt
 
+from koopmans.io import read
+
 # Load the workflow
 wf = read('cri3.kwf')
 
@@ -11,13 +12,11 @@
 ki_bs = ui_calc.results['band structure']
 ki_bs_shifted = ki_bs.subtract_reference()
 
-
 # Plot the two band structures
-ax = ki_bs_shifted.plot(spin=0, label='KI@LDA-up', color='tab:red', ls="-")
-ax = ki_bs_shifted.plot(ax=ax, spin=1, label='KI@LDA-down', color='tab:red', ls="--")
+ax = ki_bs_shifted.plot(spin=0, label='spin up', color='tab:red', ls="-")
+ax = ki_bs_shifted.plot(ax=ax, spin=1, label='spin down', color='tab:blue', ls="-")
 
 ax.legend(loc='lower right', ncol=2, bbox_to_anchor=(1, 1))
 ax.set_ylim([-5, 5])
 
 plt.savefig('cri3_bandstructures.png')
-