从满足某些条件的两个 Matlab 向量中提取元素的位置

Question

考虑 Matlab 中的三个行向量，A、B、C，每个向量的大小为1xJ。我想构造一个矩阵D，大小为Kx3，列出每个三元组(a,b,c)，这样：

• a 是A(a) 在A 中的位置。

• b 是B(b) 在B 中的位置。

• A(a)-B(b) 是C 的一个元素。

• c 是A(a)-B(b) 在C 中的位置。

• A(a) 和 B(b) 与 Inf、-Inf 不同。

例如，

A=[-3 3 0 Inf -Inf];
B=[-2 2 0 Inf -Inf];
C=[Inf -Inf -1 1 0];
D=[1 1 3;   %-3-(-2)=-1
   2 2 4;   % 3-2=1
   3 3 5];  % 0-0=0

我希望这段代码高效，因为在我的实际示例中，我必须重复多次。

这个问题和我之前的问题here有关，但是现在我正在寻找元素的位置。

Answer 1

您可以使用combvec（或any number of alternatives）获取对应数组A和B的所有索引a和b配对。那么这只是遵循您的标准的情况

1. 找出差异
2. 检查C中的差异
3. 删除您不关心的元素

像这样：

% Generate all index pairings
D = combvec( 1:numel(A), 1:numel(B) ).';
% Calculate deltas
delta = A(D(:,1)) - B(D(:,2));
delta = delta(:); % make it a column
% Get delta index in C (0 if not present)
[~,D(:,3)] = ismember(delta,C);
% If A or B are inf then the delta is Inf or NaN, remove these
idxRemove = isinf(delta) | isnan(delta) | D(:,3) == 0;
D(idxRemove,:) = [];

对于您的示例，这会产生问题的预期结果。

您说 A 和 B 最多有 7 个元素，因此您最多需要检查 49 个配对。这还不算太糟糕，但读者应该小心，对于更大的输入，配对可能会快速增长。