如何在新矩阵中循环和保存值

Question

我有以下代码，可识别KK不同值的速度值。

load('Vq.mat')
KK=1;
ft = fftshift(fft2(Vq));
kx = [-70:1:70];
ky = [-70:1:70];
for i1=1:length(kx)
    for i2=1:length(ky)
        kh(i1,i2)=sqrt(kx(i1).^2+ky(i2).^2);
        if (kh(i1,i2)>KK && kh(i1,i2)<KK+1)

        else
            ft(i1,i2)=0;
        end

    end
end
K=sum(ft);

因此，当我将KK的值设置为 1 时，代码将循环遍历矩阵kh并找到KK = 1的位置并获取该位置的速度（通过 if 语句）。 然后代码将ft矩阵中的所有其他值归零，并且仅将速度值保持在KK = 1 。 然后我将这些值相加以创建K 。

我的问题是我想从1:70循环遍历KK值。 因此，当KK = 2时，代码将再次循环遍历kh并识别KK = 2处的速度。 然后它将这些值相加，然后添加到矩阵K中。

最后，K 的单个矩阵应包括在不同 KK 值处找到的各个值的所有总和

https://www.dropbox.com/s/vi6un5jyby7akej/Vq.mat?dl=0这是数据文件的链接。

Answer 1

下面的代码被大大简化了。 主要思想是，您使用find内置 function 找到KK和KK+1之间的kh值的索引 ( inds )，并将相同索引处的相应ft值相加。 您对KK的所有值重复此算法并添加所有计算的总和。

load('Vq.mat')
ft = fftshift(fft2(Vq));
kx = -70:70;
ky = -70:70;
kh = sqrt(kx.^2 + ky.^2.'); % compute kh only once

sumK = 0;       % accumulate sum for each KK
for KK = 1:70   % loop over all kk values
    inds = find(kh > KK & kh < KK+1);
    sumK = sumK + sum(ft(inds));
end
disp(sumK)      % 267.85 - 2.8089e-14i

看看矢量化如何简化代码并使其运行得更快。 例如，我只在循环外创建了一次kh矩阵，而不是计算 70 次。 与if条件和for循环相比，还使用诸如find之类的矢量化内置函数极大地提高了代码的性能和可读性。 为了简洁和速度，应该编写惯用的 MATLAB 代码。

更新：这是 OP 评论后的更新答案：

load('Vq.mat')
ft = fftshift(fft2(Vq));
kx = -70:70;
ky = -70:70;
kh = sqrt(kx.^2 + ky.^2.'); % compute kh only once

K = zeros(70,141);    % accumulate sum for each KK
for KK = 1:70         % loop over all kk values
    K(KK,:) = sum( ft .* (kh > KK & kh < KK+1) );
end

这就是K最后的样子（ spy(K) ）：