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

12 votes130 views6 answers
Print variable python
Python
2 votes135 views1 answer
Pandas convert string from INT TO str
Python
10 votes119 views3 answers
Count occurrence in array python
Python
0 vote125 views1 answer
Getting range lowest and highest values from np array
Python
0 vote126 views1 answer
Dictionary with list as value py
Python
0 vote172 views1 answer
Listing of django model types
Python Django
0 vote122 views1 answer
Run exe for python and wait until finish
Python
1 vote127 views1 answer
Convert tibble to dataframe
Python
0 vote88 views1 answer
Sql to python converter
Python
0 vote149 views1 answer
Sending email in django
Python Django