python-矩阵内某个小矩阵的替换

1、可以使用切片操作:

import numpy as np
a = np.arange(20).reshape((4,5))
b = np.array([-1, -1, -1, -1]).reshape((2, 2))
a[0:2, 0:2] = b     #切片操作
print(a)

2、或者类似于坐标操作:

import numpy as np

#原始矩阵
mat = np.arange(16).reshape(4, 4)
print(mat)

#操作的行范围  [0~1]
rows = np.arange(2)[:, np.newaxis]

#操作的列范围  [0~1]
cols = np.arange(2)

#替换矩阵
target_mat = np.array([-1, -2, -3, -4]).reshape(2, 2)

#进行替换
mat[rows, cols] = target_mat

#替换后的矩阵
print(mat)

相当于创建了一个替换用的位置矩阵:
[ (0,0) ( 0,1)
(1,0) (1,1)]
将原始矩阵这个位置的元素进行了替代

结果:
python-矩阵内某个小矩阵的替换_第1张图片

你可能感兴趣的:(python随笔,python)