鉴于以下输入 numpy 2d-arrayA
可以通过文件通过以下链接hill_mat.npy
检索,如果我可以使用像scipy.sparse.linalg.eigs这样的迭代求解器仅计算其特征值的一个子集,那就太好了。
首先,一点上下文。该矩阵A
由大小的二次特征值问题产生,N
该问题已在双倍大小的等效特征值问题中线性化2*N
。A
具有以下结构(蓝色为零):
plt.imshow(np.where(A > 1e-15,1.,0), interpolation='None')
以及以下特点:
A shape = (748, 748)
A dtype = float64
A sparsity ratio = 77.64841716949297 %
的真实尺寸A
比这个可重复的小例子大得多。95%
我希望真实的稀疏率和形状接近(5508, 5508)
这种情况。
得到的特征值A
是复数(以复共轭对的形式出现),我对模数虚部最小的特征值更感兴趣。
问题:使用直接求解器时:
w_dense = np.linalg.eigvals(A)
idx = np.argsort(abs(w_dense.imag))
w_dense = w_dense[idx]
计算时间迅速变得令人望而却步。因此,我希望使用稀疏算法:
from scipy.sparse import csc_matrix, linalg as sla
w_sparse = sla.eigs(A, k=100, sigma=0+0j, which='SI', return_eigenvectors=False)
但似乎 ARPACK 没有以这种方式找到任何特征值。从scipy/arpack 教程中,当寻找像 的小特征值时,应该通过指定kwargwhich = 'SI'
来使用所谓的移位反转模式,即为了让算法知道它可以在哪里找到这些特征值。尽管如此,我所有的尝试都没有产生任何结果......sigma
是否有对此功能更有经验的人帮我完成这项工作?
下面是一个完整的代码片段:
import numpy as np
from matplotlib import pyplot as plt
from scipy.sparse import csc_matrix, linalg as sla
A = np.load('hill_mat.npy')
print('A shape =', A.shape)
print('A dtype =', A.dtype)
print('A sparsity ratio =',(np.product(A.shape) - np.count_nonzero(A)) / np.product(A.shape) *100, '%')
# quick look at the structure of A
plt.imshow(np.where(A > 1e-15,1.,0), interpolation='None')
# direct
w_dense = np.linalg.eigvals(A)
idx = np.argsort(abs(w_dense.imag))
w_dense = w_dense[idx]
# sparse
w_sparse = sla.eigs(csc_matrix(A), k=100, sigma=0+0j, which='SI', return_eigenvectors=False)