如何将向量(沿第四维)分配给预先指定的下标

时间:2019-01-16 16:39:04

标签: matlab matrix variable-assignment

我有一个x,y,z矩阵坐标矩阵以及一个与这些坐标相关的时间序列相对应的行向量矩阵。例如

coordinates = [1 2 3; 57 89 22]; % Where the column 1 = x, column 2 = y, column 3 = z
timeseries = rand(2,200); % where each row corresponds to the timeseries of the coordinates in the same row in the coordinates matrix.

我想构建一个包含这些时间序列的4D矩阵。任何未分配的坐标应默认为零向量。目前,我的操作如下:

M = zeros(100,100,100,200); 
for ii = 1:size(coordinates,1)
    M(coordinates(ii,1),coordinates(ii,2),coordinates(ii,3),:) = timeseries(ii,:);
end

这行得通,但是我想知道是否有一种(更具可读性/效率)的方法可以一步编写for循环。我已经尝试过使用逻辑数组和索引,但是它总是失败,因为我是在分配向量,而不是标量。

1 个答案:

答案 0 :(得分:2)

这是使用sub2ind的一种方法。我还没有计时:

sz = [100 100 100 200];
M = zeros(sz(1)*sz(2)*sz(3), sz(4));
ind = sub2ind(sz([1 2 3]), coordinates(:,1), coordinates(:,2), coordinates(:,3));
M(ind(:),:) = timeseries;
M = reshape(M, sz);

您可以通过手动计算来稍微代替sub2ind来提高速度:

sz = [100 100 100 200];
M = zeros(sz(1)*sz(2)*sz(3), sz(4));
ind = coordinates(:,1) + sz(1)*(coordinates(:,2)-1) + sz(1)*sz(2)*(coordinates(:,3)-1);
M(ind(:),:) = timeseries;
M = reshape(M, sz);