在 MATLAB 中从数组中选择元素

我知道在 MATLAB 中，在一维情况下，您可以选择具有索引的元素，例如a([1 5 3])，返回 a 的第 1 个、第 5 个和第 3 个元素。我有一个二维数组，并且想根据我拥有的一组元组选择单个元素。所以我可能想要得到a(1,3), a(1,4), a(2,5)等等。目前我拥有的最好的是diag(a(tuples(:,1), tuples(:,2))，但这需要大量的内存来存储较大的 a 和/或元组。我是否必须将这些元组转换为线性索引，或者是否有一种更清晰的方法可以在不占用太多内存的情况下完成我想要的任务？

转换为线性索引似乎是一种合法的方法：

indices = tuples(:, 1) + size(a,1)*(tuples(:,2)-1);
selection = a(indices);

请注意，这也在 Matlab 内置解决方案中实现sub2ind，如 nate'2 答案中所示：

a(sub2ind(size(a), tuples(:,1),tuples(:,2)))

然而，

a = rand(50);
tuples = [1,1; 1,4; 2,5];

start = tic;
for ii = 1:1e4
    indices = tuples(:,1) + size(a,1)*(tuples(:,2)-1); end
time1 = toc(start);


start = tic;
for ii = 1:1e4
    sub2ind(size(a),tuples(:,1),tuples(:,2)); end
time2 = toc(start);

round(time2/time1)

这使

ans =   
    38

所以虽然sub2ind眼睛更舒服，也是~40倍slower。如果您必须经常进行此操作，请选择上面的方法。否则，使用sub2ind以提高可读性。