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

-1 vote287 views3 answers
Transpose of a matrix in python numpy
Python
0 vote370 views1 answer
How to export schema in graphene django
Python Django
0 vote321 views1 answer
Django create token for user
Python Django
5 votes286 views2 answers
Jupyter nbextension
Python
2 votes299 views2 answers
Pd.dataframe initial columns
Python
12 votes309 views2 answers
Python input integer only
Python
4 votes314 views2 answers
Loginrequiredmixin django
Python Django
3 votes279 views2 answers
Python list comma separated string
Python
-1 vote279 views1 answer
Significant figures on axes plot matplotlib
Python
0 vote238 views1 answer
Collecting pipnev
Python