https://docs.scipy.org/doc/scipy/reference/generated/scipy.sparse.dia_matrix.html dia_matrix((data, offsets), shape=(M, N)) where the data[k,:] stores the diagonal entries for diagonal offsets[k] (See example below)
这个真是太难懂了。。
只好搞个例子。。。
data = np.array([[1,2,3,4],
[5,6,7,8],
[9,10,11,12]])
print(data)
offsets = np.array([0, -1, 1])
sp.dia_matrix((data, offsets), shape=(4, 4)).toarray()
# output
[[ 1 2 3 4]
[ 5 6 7 8]
[ 9 10 11 12]]
array([[ 1, 10, 0, 0],
[ 5, 2, 11, 0],
[ 0, 6, 3, 12],
[ 0, 0, 7, 4]])
- k = 0, data[k, :] = [1,2,3,4], offsets[k] = 0,则主对角线上的数值是 data[k, :].
- k = 1, data[k, :] = [5,6,7,8], offsets[k] = -1,则主对角线下面第一个斜对角线上的的数值是 data[k, :],并且从 data[k, :][0] 开始从左上到右下正序排列,超出目标矩阵的部分会截掉。
- k = 2, data[k, :] = [9,10,11,1], offsets[k] = 1,则主对角线下面第一个斜对角线上的的数值是 data[k, :],并且从 data[k, :][-1] 开始从右下到左上倒序排列,超出目标矩阵的部分会截掉。这里需要注意,右边界不会超出原矩阵的宽度。(注意看下面这个例子中的 [10,11,12] 是怎么排列的。)
data = np.array([[1,2,3,4],
[5,6,7,8],
[9,10,11,12]])
print(data)
offsets = np.array([0, -5, 1])
sp.dia_matrix((data, offsets), shape=(10, 5)).toarray()
# output
[[ 1 2 3 4]
[ 5 6 7 8]
[ 9 10 11 12]]
array([[ 1, 10, 0, 0, 0],
[ 0, 2, 11, 0, 0],
[ 0, 0, 3, 12, 0],
[ 0, 0, 0, 4, 0],
[ 0, 0, 0, 0, 0],
[ 5, 0, 0, 0, 0],
[ 0, 6, 0, 0, 0],
[ 0, 0, 7, 0, 0],
[ 0, 0, 0, 8, 0],
[ 0, 0, 0, 0, 0]])