arrays - 在 Matlab 中向量化和分离查找操作

Question

我将尝试通过一个例子来解释我正在寻找的东西。

假设我有一个像这样的 3x4 2D 矩阵:

1 0 0 0
1 1 1 0
0 0 1 1

这是可以通过以下方式生成的 matlab 矩阵:

x = [1 0 0 0; 1 1 1 0; 0 0 1 1]

现在如果我执行这样的命令:

[~, y] = find(x([1 2], :) == 1)

y 将是一个向量，其中包含第 1 行和第 2 行中值为 1 的列索引。具体而言，对于此示例:

y = [1 1 2 3]

但是，我希望完成的是将输入数组(此处为 [1 2])的每个元素中的非零列分开，而不必迭代此输入数组的元素(以矢量化方式)。

所以我希望得到这样的输出:

y = [[1] [1 2 3]]

不严格采用这种格式，而是采用任何分隔形式。

Answer 1

如果我没理解错你的问题，你想根据行对其进行分组。然后维护行:

[r, y] = find(x([1 2], :) == 1)

然后你可以使用accumarray

y2 = accumarray(r,y,[max(r),1],@(x)({x}))

对于你的例子，它返回

y2 = {[1],[1 2 3]}