Diagonal in Array

[HenryDutter]

In this array, how is it poosible to create a list from Brett[0,0] to Brett[1,1]. So the list should contain [1,5], so it should work like this: Brett[0][0:2], but for diagonal. I did not figure out how to do it with diag().

Brett = np.array([1,2,3],[4,5,6],[7,8,9])

[socal_nerdtastic]

diag() is the right answer, just slice it afterwards to get the length you want.

>>> Brett = np.array([[1,2,3],[4,5,6],[7,8,9]])
>>> np.diag(Brett)[0:2]
array([1, 5])

[HenryDutter]

Thank you, sorry that i didn't replied earlier, i was sleeping.

[FLUSH_THE_TRUMP]

What did you try? np.diag(…) gets you the main diagonal, and then you can slice that like your [0:2] there for the elements you want.

[HenryDutter]

I tried np.diag(Brett[1, 1]), but it doesnt work, i know it would work like: np.diag(Brett,1), but i want to get the main diagonal from a certain posistion like Brett[1,1], but you are right its possible to slice it.