将矢量拆分为预定义大小的组

Question

我正在寻找一种方法，将输入向量拆分为预定义大小的组，如果有余数，最后一组会更小。我更喜欢将输出作为一个单元格，但我不介意它是否是任何其他类，只要它使用后续索引提供对子组的访问。

以下是预期行为的示例：

% Even split
v = 1:6;
grpSz = 2;
% OUT: {[1,2], [3,4], [5,6]}
% Remainder
v = 1:5;
grpSz = 3;
% OUT: {[1,2,3], [4,5]}
% Single group
v = 1:5;
grpSz = 6;
% OUT: {[1,2,3,4,5]}

以下是几个有用的数量：

• 输出中的组数：nG = ceil(numel(v)/grpSz)
• 最后一组中的元素数：r = mod(numel(v), grpSz)

---

目前我正在使用mat2cell:

function out = evenSplitter(v, grpSz)
nV = numel(v);
nG = ceil(nV/grpSz);
r = mod(nV, grpSz);
out = mat2cell(v, 1, [repmat(grpSz, 1, nG-1), ~r*grpSz+r]);

这很有效，但看起来有点笨重。有人会建议一个更优雅的解决方案吗？

Answer 1

一个使用splitapply:的解决方案

v     = 1:5;                                           % input array
grpSz = 2;                                             % maximal group size
out   = splitapply(@(x){x},v,ceil((1:numel(v))/grpSz)) % split v

这个方法适用于所有给定的例子。

splitapply将数据分组并应用函数。匿名函数@(x){x}只是将一个组的每个元素放入一个单元格中。并且ceil((1:numel(v))/grpSz)创建一个数组，该数组指示哪些元素与哪些组相关联。

例如，如果v = 1:5和grpSz = 2，则ceil((1:numel(v))/grpSz)生成以下数组[1 1 2 2 3]。

Answer 2

对于大输入，这可能比accumarray更快：

out = arrayfun(@(k) v(k:min(k+grpSz-1, end)), 1:grpSz:numel(v), 'UniformOutput', false);

Answer 3

另一个解决方案，受到obchardon答案的启发(在很大程度上相当于(：

function out = evenSplitter(v, grpSz)
idx = ceil((1:numel(v))/grpSz);                       % Group ID of each element
out = accumarray( idx.', v, [idx(end) 1], @(x){x} );  % idx(end) == nG

这使用可能比splitapply更快的accumarray。