Numpy diag()

>>> x = np.arange(9).reshape((3,3))
>>> x
array([[0, 1, 2],
       [3, 4, 5],
       [6, 7, 8]])

>>> np.diag(x)
array([0, 4, 8])
>>> np.diag(x, k=1)
array([1, 5])
>>> np.diag(x, k=-1)
array([3, 7])

Source: numpy.org

Similar codes

0 vote186 views1 answer
Free wifi connection disconnects frequently windows 10
Python
0 vote212 views1 answer
Numpy savetxt list of strings
Python
-1 vote177 views1 answer
How to print memory address in python
Python
0 vote196 views1 answer
How to create datetime from negative epoch in python
Python
1 vote196 views1 answer
Pros and cons of python flush print function
Python
0 vote224 views1 answer
Python why is it important to check the __name__
Python
0 vote281 views1 answer
Qt line edit set text python
Python
0 vote212 views1 answer
Get array from h5py dataset
Python
0 vote204 views1 answer
Python copy file create intermediate directories
Python
0 vote175 views1 answer
Add a title to pandas dataframe
Python