有没有人可以用Python快速解决这个问题。在矩阵 A 中,我希望在某个位置用 C 矩阵替换 A 中的值?
A= [ 1 2 3 4 4
2 3 4 5 3
4 5 6 7 2
5 6 7 8 9
6 7 8 9 0 ]
C= [ 7 7
7 7 ]
在位置loc = [3 5],使得结果矩阵 B 如下所示:
B= [ 1 2 3 4 4
2 3 4 5 3
4 5 7 7 7
5 6 7 8 9
6 7 7 9 7 ]
在 Matlab 中,代码很简单:
A(loc,loc) = C
我试过它是Python:
A[loc,loc] = C
没有成功。有任何想法吗?任何使用局部刚度模型填充全局刚度矩阵的人都会多次执行该操作,但通常看起来像这样:
K(loc,loc) = K(loc,loc) + C
无论如何,我希望这是一个好的问题?我还没有找到答案。
最佳答案
import numpy as np
A = np.array([ 1, 2, 3, 4, 4,
2, 3, 4, 5, 3,
4, 5, 6, 7, 2,
5, 6, 7, 8, 9,
6, 7, 8, 9, 0 ]).reshape(5, 5)
B = np.array([60, 70, 80, 90]).reshape(2, 2)
A[2:6:2, 2:6:2] = B # or
A[np.ix_([2, 4], [2, 4])] = B
print A
输出:
[[ 1 2 3 4 4]
[ 2 3 4 5 3]
[ 4 5 60 7 70]
[ 5 6 7 8 9]
[ 6 7 80 9 90]]
关于Python Numpy - 矩阵替换全局矩阵中定义的矩阵..类似于 Matlab,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/20000185/