Matlab中无需循环实现数组成员判断及代码简化技术咨询

完全可以实现无循环的版本！先拆解一下你原代码的核心逻辑：你要构建一个对角矩阵PID，其中第i个对角元素等于稀疏矩阵TFP第i行中，列索引j属于集合A的所有元素之和，非对角元素保持初始值（默认应该是0）。

这里有个高效的向量化解法，还能保留TFP的稀疏特性：

% 生成列索引的掩码：标记哪些j属于A
mask = ismember(1:N, A);
% 提取TFP中符合条件的列，每行求和后转为对角矩阵
PID = diag(sum(TFP(:, mask), 2));

代码解释 #

• ismember(1:N, A)：生成一个长度为N的逻辑向量，其中mask(j)为true当且仅当j是A中的元素，完美对应你原代码里的ismember(j,A)判断。
• TFP(:, mask)：直接从稀疏矩阵TFP中提取所有列索引在A里的子矩阵，稀疏矩阵的切片操作效率很高，不会额外占用太多内存。
• sum(TFP(:, mask), 2)：对提取后的子矩阵按行求和，得到一个N×1的向量，每个元素就是原循环中PID(i,i)的累加结果。
• diag(...)：把这个求和向量转换成对角矩阵，和原循环生成的PID结构完全一致——非对角元素为0，对角元素是对应行的符合条件的和。

特殊情况处理 #

如果你的PID原本有初始值（不是全0矩阵），需要在原有基础上累加的话，只需要改成：

PID = PID + diag(sum(TFP(:, mask), 2));

这个方法完全避开了嵌套循环，利用Matlab的矩阵化操作优势，尤其是对于稀疏矩阵来说，效率会比循环高很多哦！

内容的提问来源于stack exchange，提问作者dtr43