scipy.sparse.dia_matrix 中的 offsets 含义

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]])
​
  • 3
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值