Browse Source

Minor revisions

master
Karthik 2 years ago
parent
commit
58f51a6172
  1. 16
      DyLevelStructure/visualizeDyLevelStructureWithTransitions.py

16
DyLevelStructure/visualizeDyLevelStructureWithTransitions.py

@ -117,9 +117,9 @@ def plot_level_structure_with_red_and_blue_transitions(*args, **kwargs):
#figure options #figure options
f.canvas.draw() 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)) plot_handle.set_xticks(range(min_J-1, max_J+2))
ax.get_xticklabels()[0].set_visible(False) 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 = [item.get_text() for item in ax.get_yticklabels()]
yticklabels = ['' if item.startswith('') or item.startswith('0') else item for item in 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] 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) plot_handle.set_yticks(new_yticks)
new_yticklabels = [round(1e7/item) if item != 0 else item for item in new_yticks] new_yticklabels = [round(1e7/item) if item != 0 else item for item in new_yticks]
ax.set_yticklabels(new_yticklabels) ax.set_yticklabels(new_yticklabels)
ax.get_yticklabels()[0].set_visible(False) 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()[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) plt.tick_params(axis='both', which='major', labelsize=14)
@ -149,7 +149,7 @@ if __name__ == '__main__':
min_J = 8 min_J = 8
max_J = 9 max_J = 9
max_wavenumber = 25500.0
max_wavenumber = 24500.0
#Ground State: [Xe]4f^{10} 6s^2(^5I_8) #Ground State: [Xe]4f^{10} 6s^2(^5I_8)
gs_J = 8 gs_J = 8
@ -167,7 +167,7 @@ if __name__ == '__main__':
home_path = str(Path(__file__).parent.resolve()) 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) 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)
Loading…
Cancel
Save