scipy.sparse.

eye_array#

scipy.sparse.eye_array(m, n=None, *, k=0, dtype= 'float'>, format=None)[источник]#

Разреженный массив заданной формы с единицами на k-й диагонали и нулями в остальных местах.

Возвращает разреженный массив с единицами на диагонали. Конкретно разреженный массив (m x n), где k-я диагональ состоит из единиц, а все остальное - нули.

Параметры:
mint

Запрошенное количество строк.

nint, необязательный

Количество столбцов. По умолчанию: m.

kint, необязательный

Диагональ для размещения единиц. По умолчанию: 0 (главная диагональ).

dtypedtype, опционально

Тип данных массива

форматstr, optional (по умолчанию: “dia”)

Разреженный формат результата, например, format=”csr” и т.д.

Возвращает:
new_arrayразреженный массив

Разреженный массив заданной формы с единицами на k-й диагонали и нулями в остальных местах.

Примеры

>>> import numpy as np
>>> import scipy as sp
>>> sp.sparse.eye_array(3).toarray()
array([[ 1.,  0.,  0.],
       [ 0.,  1.,  0.],
       [ 0.,  0.,  1.]])
>>> sp.sparse.eye_array(3, dtype=np.int8)

    with 3 stored elements (1 diagonals) and shape (3, 3)>