scipy.sparse.spdiags

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

从对角线返回稀疏矩阵。

参数
dataarray_like

按行存储的矩阵对角线

diags要设置的对角线
  • K=0主对角线

  • k>0的第k条上对角线

  • k<0第k下对角线

m, n集成

结果的形状

format字符串,可选

结果的格式。默认情况下(FORMAT=NONE)返回适当的稀疏矩阵格式。这一选择可能会发生变化。

参见

diags

此函数的更方便形式

dia_matrix

稀疏对角线格式。

示例

>>> 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]])