Octave function 获取矩阵中的连续列组

Question

我试图找到一种有效的方法来提取矩阵中的n个连续列组。 例子：

A = [0, 1, 2, 3, 4; 0, 1, 2, 3, 4; 0, 1, 2, 3, 4]; 
n = 3;

应该产生一个与此类似的 output：

answer = cat(3, ...
                [0, 1, 2; 0, 1, 2; 0, 1, 2], ...
                [1, 2, 3; 1, 2, 3; 1, 2, 3], ...
                [2, 3, 4; 2, 3, 4; 2, 3, 4]);

我知道这可以使用for循环来实现，例如以下代码片段：

answer = zeros([3, 3, 3]);
for i=1:3
  answer(:, :, i) = A(:, i:i+2);
endfor

但是，我试图避免在这种情况下使用for循环 - 是否也有可能将此操作向量化（使用索引表达式）？

Answer 1

仅使用索引

ind = reshape(1:size(A,1)*n, [], n) + reshape((0:size(A,2)-n)*size(A,1), 1, 1, []); 
result = A(ind);

索引ind是使用线性索引和隐式扩展构建的。

使用图像 Package / 图像处理工具箱

result = reshape(im2col(A, [size(A,1) n], 'sliding'), size(A,1), n, []);

这里的大部分工作是由带有'sliding'选项的im2col function 完成的。