Home - News ( United States | United Kingdom | Italy | Germany ) - Football scores

Showing content from http://docs.scipy.org/doc/numpy/reference/generated/numpy.diag_indices_from.html below:

numpy.diag_indices_from — NumPy v2.3 Manual

numpy.diag_indices_from(arr)[source]#

Return the indices to access the main diagonal of an n-dimensional array.

See diag_indices for full details.

Parameters:

arrarray, at least 2-D

Examples

Create a 4 by 4 array.

>>> a = np.arange(16).reshape(4, 4)
>>> a
array([[ 0,  1,  2,  3],
       [ 4,  5,  6,  7],
       [ 8,  9, 10, 11],
       [12, 13, 14, 15]])

Get the indices of the diagonal elements.

>>> di = np.diag_indices_from(a)
>>> di
(array([0, 1, 2, 3]), array([0, 1, 2, 3]))

>>> a[di]
array([ 0,  5, 10, 15])

This is simply syntactic sugar for diag_indices.

>>> np.diag_indices(a.shape[0])
(array([0, 1, 2, 3]), array([0, 1, 2, 3]))

RetroSearch is an open source project built by @garambo | Open a GitHub Issue

Search and Browse the WWW like it's 1997 | Search results from DuckDuckGo

HTML: 3.2 | Encoding: UTF-8 | Version: 0.7.4