Julia中如何不创建新数组统计满足条件的矩阵对角元数量？

可以通过生成器表达式或直接遍历对角索引的方式，完全避免创建中间数组，实现统计需求：

方法1：使用count结合生成器 #

threshold = 0.1
count(i -> A[i,i] < threshold, 1:min(size(A)...))

原理：

• min(size(A)...)获取矩阵行数和列数的较小值，确保遍历范围仅覆盖实际存在的对角元（适配非方阵场景）
• 生成器i -> A[i,i] < threshold直接逐个访问对角元并判断条件，不会生成任何中间数组

方法2：使用sum结合布尔生成器 #

threshold = 0.1
sum(A[i,i] < threshold for i in 1:min(size(A)...))

原理：

• Julia中布尔值true会被视为1，false视为0，因此对布尔条件求和等价于统计满足条件的元素数量
• 同样通过生成器惰性遍历对角元，无中间数组开销

补充：使用视图（无数据复制） #

如果需要对对角元进行更多操作，也可以用view创建对角元的视图（仅引用原矩阵数据，不复制），再统计：

using LinearAlgebra
threshold = 0.1
count(<(threshold), view(A, diagind(A)))

diagind(A)返回对角元的索引集合，view基于这些索引创建原矩阵的视图，不会生成新数组。

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