matlab - 使用向量元素选择矩阵元素

Question

这个问题已经有答案了:

Use a vector as an index to a matrix (3 个回答)

已关闭 4 年前。

Matlab 有没有办法根据向量中的元素选择矩阵元素？我认为我的描述不清楚，但我实际上想做的是类似于:

A=zeros(3,3,3) %3d matrix
A(1,1,2)=5
b=[1,1,2]
A(b)=5

意思是，使用向量中的条目作为参数从矩阵中选择一个元素的一种简单方法。这个确切的示例不起作用，因为最后一行将 b 视为单个参数，而不是三个。我可以写 A(b(1),b(2),b(3)) 但我真正在这里寻找的是是否有一种好的方法。

Answer 1

方法一:使用sub2ind找到线性索引

您可以定义一个名为 findLinearIndex 的函数，以便将向量元素转换为 A 的线性索引:

findLinearIndex = @(A,b) sub2ind(size(A), b(1), b(2), b(3))
A(findLinearIndex(A,b)) = 5

方法二:通过 num2cell 将向量转换为元胞数组

然后，您可以使用{:}来获取索引

b_cell = num2cell(b) ;
A(b_cell{:}) = 5