import numpy as np
nlevels = 3
nmicrostates = 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))
nmicrostates += 1
print('Number of microstates: ' + str(nmicrostates))
[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.]
Number of microstates: 9