import numpy as np
nlevels = 3
total_microstates = 0
for p1 in range(nlevels):
for p2 in range(nlevels):
iconfig = np.zeros(nlevels)
iconfig[p1] += 1
iconfig[p2] += 1
print('%s' % str(iconfig))
total_microstates += 1
print("The total number of microstates is " + str(total_microstates) + ".")
[2. 0. 0.]
[1. 1. 0.]
[1. 0. 1.]
[1. 1. 0.]
[0. 2. 0.]
[0. 1. 1.]
[1. 0. 1.]
[0. 1. 1.]
[0. 0. 2.]
The total number of microstates is 9.