给出矩阵A,行索引列表和列索引列表,如何有效地提取以行索引和列索引为中心的大小为k的平方子矩阵?
例如:
A = array([[12,6,14,8,4,1],[18,13,10,9,19],[ 8,15,5,18],[ 3,2,12],[ 4,19,14],[16,7,11,0],5],1,12,12]])
r = np.array([2,5])
c = np.array([3,2])
k = 3
输出应为A[1:4,2:5]
和A[4:7,1:4]
。因此,基本上,输出是大小为kxk
的平方子矩阵,并以[r,c]元素(在这种情况下为A [2,3]和A [5,2])为中心
如何高效而优雅地做到这一点?谢谢