基于一维索引列表有效地将值分配给多维数组

问题描述：

我有一个大小为[S1, S2, S3]的矩阵M。基于一维索引列表有效地将值分配给多维数组

我有另一个矩阵K作为我想分配的第一个维度的索引，大小为[1, S2, S3]。

And V是一个[1, S2, S3]矩阵，它包含要相应分配的值。

随着循环，这是我做的：

for x2 = 1:S2 
    for x3 = 1:S3 
    M(K(1,x2,x3), x2, x3) = V(1, x2, x3) 
    endfor % x3 
endfor % x2

有没有更有效的方式来做到这一点？

可视化2D情况：

 
M = 
    1 4 7 10 
    2 5 8 11 
    3 6 9 12 

K = 
    2 1 3 2 

V = 
    50 80 70 60 

Desired = 
    1 80 7 10 
50 5 8 60 
    3 6 70 12

测试用例：

 
M = reshape(1:24, [3,4,2]) 
K = reshape([2,1,3,2,3,3,1,2], [1,4,2]) 
V = reshape(10:10:80, [1,4,2]) 
s = size(M) 
M = assign_values(M, K, V) 
M = 

ans(:,:,1) = 

    1 20 7 10 
    10 5 8 40 
    3 6 30 12 

ans(:,:,2) = 

    13 16 70 22 
    14 17 20 80 
    50 60 21 24

我在寻找实现assign_values有一种有效的方式。

莫名其妙运行Gelliant的回答给了我这样的：

 
key = sub2ind(s, K, [1:s(2)]) 
error: sub2ind: all subscripts must be of the same size

答

您可以使用sub2ind使用您的个人标线性指标。然后可以用这些值替换V中的值。

M = [1 4 7 10 ;... 
    2 5 8 11 ;... 
    3 6 9 12]; 
s=size(M); 
K = [2 1 3 2]; 
K = sub2ind(s,K,[1:s(2)]) 
V = [50 80 70 60]; 
M(K)=V;

你不需要reshape和M=M(:)为它在Matlab工作。

而不是'K = sub2ind（s，K，[1：s（1）]）'它一定是'K = sub2ind（s，K，[1：s（2）]'， – Irreducible

谢谢，我更新了我的答案。 – Gelliant

嗯，它说'所有的下标必须是相同的大小'（我正在使用八度）。 – justhalf

答

我发现这工作：

K = K(:)'+(S1*(0:numel(K)-1)); 
M(K) = V;

或许这是应该的工作方式Gelliant的答案一样，但我不能让他的答复工作，不知何故=/

基于一维索引列表有效地将值分配给多维数组

相关推荐