这是一个非常简单的问题,但我找不到任何解决方案。
如果我有一个矩阵
A = np.array([[1,2,4,2],[1,2,35,4],[3,4,7,0],[8,3,2,6]])
我想提取由第一、第二和第四列和第一、第二和第四行
组成的子矩阵在 MATLAB 中,我只需执行 A([1,2,4],[1,2,4])
,但在 Python 中无法执行此操作。
有什么快速的方法吗?我需要避免 for 循环,因为我正在处理有限元代码。
最佳答案
您的问题的解决方案是np.ix_
:
A[np.ix_([0,1,3], [0,1,3])]
输出:
array([[1, 2, 2],
[1, 2, 4],
[8, 3, 6]])
关于python - Python中的特定子矩阵提取,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/59000671/