sparse.diags

sparse.diags是Scipy中的一个函数，用于创建稀疏矩阵对角线。它的输入是一个列表，其中每个元素都是一个向量，表示矩阵的每个对角线。函数还接受一些可选参数，例如偏移量和输出格式。下面是一个简单的例子，创建一个3x3的稀疏矩阵，其中主对角线元素为1，次对角线元素为2：

import numpy as np
from scipy.sparse import diags

# create diagonal vectors
main_diag = np.ones(3)
sub_diag = 2 * np.ones(2)

# create sparse matrix
matrix = diags([main_diag, sub_diag], [0, -1])

print(matrix.toarray())

输出结果为：

array([[1., 2., 0.],
       [0., 1., 2.],
       [0., 0., 1.]])

这是一个稀疏矩阵，其中只有5个非零元素，其他元素都是零。