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

Question

我有一个 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 循环。我试过使用逻辑数组和索引，但它总是失败，因为我分配的是向量，而不是标量。

Answer 1

这是使用 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);