用于计算矩阵元素之间边界的脚本

Question

下面的脚本给出了矩阵每个元素的边界总和。边界是关于彼此相邻且值为 1 的元素计算的。此求和的乘积命名为接触周长。

但是有没有其他方法可以总结或矢量化我的原始和简单脚本？我提出这个要求是因为我的真实矩阵非常大并且使用"for"增加了计算时间。

谢谢。

a1=[1 1 0 1 0 1;
    0 1 1 0 0 1;
    1 1 0 1 0 1;
    1 1 0 0 1 0;
    0 0 0 1 1 1]
m=5
n=6
cmp=zeros(m,n)
cmp1=zeros(m,n)
for i=1:m-1
    for j=1:n
        if a1(i,j)==a1(i+1,j) && a1(i,j)==1
            cmp(i,j)=1
        end
    end
    for i=1:m
        for j=1:n-1
            if a1(i,j)==a1(i,j+1) && a1(i,j)==1
                cmp1(i,j)=1
            end
        end
    end
end
cmtotal=cmp+cmp1
pc=sum(sum(cmtotal))

Answer 1

这应该是非常有效的 -

%// Case1 and case 2 matches
case1_matches = a1(1:end-1,:) == a1(2:end,:) & a1(1:end-1,:)==1
case2_matches = a1(:,1:end-1) == a1(:,2:end) & a1(:,1:end-1)==1

%// Get sum of those matches for the final output, equivalent to your pc
out = sum(case1_matches(:)) + sum(case2_matches(:))

您可以将 sum(..(:)) 替换为 nnz()，但根据 [，我怀疑它在运行时性能方面是否会优于 sumbenchmarks of sum against nnz.

用于计算矩阵元素之间边界的脚本

script for calculating boundaries between elements of matrix

performance

matlab

matrix

vectorization