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