当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python SciPy sparse.spdiags用法及代码示例


本文简要介绍 python 语言中 scipy.sparse.spdiags 的用法。

用法:

scipy.sparse.spdiags(data, diags, m=None, n=None, format=None)#

从对角线返回一个稀疏矩阵。

参数

data array_like

矩阵对角线按行存储

diags int 或 int 的序列

要设置的对角线:

  • k = 0 主对角线

  • k > 0 第 k 个上对角线

  • k < 0 第 k 个下对角线

m, n int、元组、可选

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

format str,可选

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

例子

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

相关用法


注:本文由纯净天空筛选整理自scipy.org大神的英文原创作品 scipy.sparse.spdiags。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。