Z = np.diag(1+np.arange(4), k=-1)
print Z
[[0 0 0 0 0]
[1 0 0 0 0]
[0 2 0 0 0]
[0 0 3 0 0]
[0 0 0 4 0]]