karthik/Plotting
1
0
Fork 0
Code Issues Pull Requests repo.project_board Releases Wiki Activity

Minor revisions

Browse Source
This commit is contained in:
Karthik 2022-12-23 20:25:15 +01:00
parent 124ef73439
commit 58f51a6172
1 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
DyLevelStructure/visualizeDyLevelStructureWithTransitions.py
View File

@ -62,7 +62,7 @@ def parse_NIST_data(path, min_J, max_J, max_wavenumber):
Parity = [ele for idx, ele in enumerate(Parity) if idx not in splice_idxs]
J = [ele for idx, ele in enumerate(J) if idx not in splice_idxs]
Wavenumber = [ele for idx, ele in enumerate(Wavenumber) if idx not in splice_idxs]
# Create a Pandas data frame with the data
dataset = pd.DataFrame(np.array(list(zip(Parity, J, Wavenumber))), columns=['Parity', 'J', 'Wavenumber'])
@ -117,9 +117,9 @@ def plot_level_structure_with_red_and_blue_transitions(*args, **kwargs):
#figure options
f.canvas.draw()
plt.xlabel('$J$', fontsize=16)
plt.ylabel('$\\tilde{v}~(cm^{-1})$', fontsize=16)
plt.ylabel('$\\lambda~(nm)$', fontsize=16)
plt.xlabel('J', fontsize=16)
#plt.ylabel('$\\tilde{v}~(cm^{-1})$', fontsize=16)
plt.ylabel('Wavelength (nm)', fontsize=16)
plot_handle.set_xticks(range(min_J-1, max_J+2))
ax.get_xticklabels()[0].set_visible(False)
@ -130,14 +130,14 @@ def plot_level_structure_with_red_and_blue_transitions(*args, **kwargs):
yticklabels = [item.get_text() for item in ax.get_yticklabels()]
yticklabels = ['' if item.startswith('−') or item.startswith('0') else item for item in yticklabels]
yticks = [float(item) if item != '' else 0.0 for item in yticklabels]
new_yticks = np.arange(min(yticks), max(yticks), 4000)
new_yticks = np.arange(min(yticks), max_wavenumber, 4000)
plot_handle.set_yticks(new_yticks)
new_yticklabels = [round(1e7/item) if item != 0 else item for item in new_yticks]
ax.set_yticklabels(new_yticklabels)
ax.get_yticklabels()[0].set_visible(False)
ax.get_yticklabels()[-1].set_visible(False)
#ax.get_yticklabels()[-1].set_visible(False)
ax.get_yticklines()[0].set_visible(False)
ax.get_yticklines()[-2].set_visible(False)
#ax.get_yticklines()[-2].set_visible(False)
plt.tick_params(axis='both', which='major', labelsize=14)
@ -149,7 +149,7 @@ if __name__ == '__main__':
min_J = 8
max_J = 9
max_wavenumber = 25500.0
max_wavenumber = 24500.0
#Ground State: [Xe]4f^{10} 6s^2(^5I_8)
gs_J = 8
@ -167,7 +167,7 @@ if __name__ == '__main__':
home_path = str(Path(__file__).parent.resolve())
dataframe = parse_NIST_data(home_path + os.sep + NIST_Dy_level_data_filename, min_J, max_J, max_wavenumber)
plot_level_structure_with_red_and_blue_transitions(dataframe, gs_J, gs_wavenumber, red_J, red_wavenumber, blue_J, blue_wavenumber, home_path)
plot_level_structure_with_red_and_blue_transitions(dataframe, gs_J, gs_wavenumber, red_J, red_wavenumber, blue_J, blue_wavenumber, max_wavenumber, home_path)