numpy diag 28 29

1>>> x = np.arange(9).reshape((3,3))
2>>> x
3array([[0, 1, 2],
4       [3, 4, 5],
5       [6, 7, 8]])
6
7>>> np.diag(x)
8array([0, 4, 8])
9>>> np.diag(x, k=1)
10array([1, 5])
11>>> np.diag(x, k=-1)
12array([3, 7])
13