ax = fig.add_subplot(1,1,1)
es = list()
for e, d in zip(*structure.getSpectrum()):
es.append(e)
ax.plot([e]*2, [0, d], color = 'blue')
ax.set_xlim(min(es)-1,max(es)+1)
plt.tight_layout()
plt.savefig('testSpectrum.pdf', dpi=300)
plt.close()
#sz_1 = .5 * (c['up', 0].H.dot(c['up', 0]) - c['dn', 0].H.dot(c['dn', 0]))
#sz_2 = .5 * (c['up', 1].H.dot(c['up', 1]) - c['dn', 1].H.dot(c['dn', 1]))
#chi_zz_tot = sz_1.dot(sz_1) + sz_2.dot(sz_2)
fig = plt.figure()
ax = fig.add_subplot(1,1,1)
energies, degeneracies = structure.getSpectrumEnergySorted()
s2s = []
datapointsDegeneracy = []
datapoints = []
for i, e, deg in zip(range(len(energies)), energies, degeneracies):
for s2 in [3*structure.getSuperpositionStatesEnergySorted()[i][j].getQuantumNumber(chi_zz_tot) for j in range(len(structure.getSuperpositionStatesEnergySorted()[i]))]:
if not (s2, e) in datapoints:
ax.scatter(s2, e, marker = '_', color = 'black')
s2s.append(s2)
datapoints.append((s2, e))
datapointsDegeneracy.append(1)
else:
k = 0
for k, datapoint in enumerate(datapoints):
if (s2, e) == datapoint:
break
