用犰狳函数替换 `sparse.eye`

Posted 2023-02-17

【中文标题】用犰狳函数替换 `sparse.eye`

【英文标题】：Replace `sparse.eye` with an armadillo function

【发布时间】：2017-02-28 09:32:51

【问题描述】：

我有一个 python 脚本，我想用犰狳用 C++ 重写。在python中我只有一行

matrix = 1/(12*h)*(sparse.eye(num_points, k = -2, dtype=np.complex).toarray() * 1 + sparse.eye(num_points, k = -1, dtype=np.complex).toarray() * -8 + sparse.eye(num_points, k = 1, dtype=np.complex).toarray() * 8 + sparse.eye(num_points, k = 2, dtype=np.complex).toarray() * -1)

这会生成一个矩阵，其中除了五个主要对角线之外的所有值都为零。不幸的是，我在犰狳中找不到类似的功能，目前我能看到的唯一方法是创建一个ones()-matrix，然后用.diag() 设置对角线，然后将其余部分归零。有没有更简单的方法来做到这一点？

【参考方案1】：

以下代码应该具有等效的功能：

// sparse matrices have all values as zero at initialization
sp_mat X(10,10);

X.diag(-2).fill( 1);
X.diag(-1).fill(-8);
X.diag(+1).fill( 8);
X.diag(+2).fill(-1);

X *= 1.0 / (12*h);  // the .0 in 1.0 tells the compiler to use the double type