scipy.sparse.

spdiags#

scipy.sparse.spdiags(data, diags, m=None, n=None, format=None)[源代码]#

从对角线返回稀疏矩阵。

参数:
data类似数组

按行存储的矩阵对角线

diagsint 序列或 int

要设置的对角线

  • k = 0 主对角线

  • k > 0 第 k 个上对角线

  • k < 0 第 k 个下对角线

m, nint,元组,可选

结果的形状。如果 n 为 None,且 m 为给定的元组,则形状为此元组。如果省略,则该矩阵是方的,其形状是 len(data[0])。

formatstr,可选

结果的格式。默认情况下(format=None),将返回合适的稀疏矩阵格式。该选项可能会更改。

.. 警告:

此函数返回一个稀疏矩阵,而非稀疏阵列。鼓励您使用 diags_array 利用稀疏阵列功能。

另请参见

diags_array

此函数的更便捷形式

diags

diags_array 的矩阵版本

dia_matrix

稀疏 DIAgonal 格式。

示例

>>> import numpy as np
>>> from scipy.sparse import spdiags
>>> data = np.array([[1, 2, 3, 4], [1, 2, 3, 4], [1, 2, 3, 4]])
>>> diags = np.array([0, -1, 2])
>>> spdiags(data, diags, 4, 4).toarray()
array([[1, 0, 3, 0],
       [1, 2, 0, 4],
       [0, 2, 3, 0],
       [0, 0, 3, 4]])