如何统计0-1方阵中非全0一维数组的数量？（Python实现）

原代码的问题 #

你的函数逻辑完全走偏了：

• 条件判断if i in MX[0 + count] == 0完全不符合预期：i就是当前遍历的行，MX[count]也是同一行，这个表达式会先计算MX[count] == 0（结果是False，因为列表不可能等于整数0），再判断i是否在False里，这永远不成立，导致所有行都会进入else分支，统计结果完全错误。
• count变量纯属多余，for循环已经在逐个遍历方阵的每一行，不需要手动维护计数变量。

正确实现方式 #

核心思路很简单：遍历方阵的每一行，判断该行是不是不全是0，统计符合条件的行数就行。

方法1：基础循环写法 #

适合新手理解的逐行检查方式：

def xArrayCount(MX):
    x = 0
    for row in MX:
        # 检查当前行有没有1
        has_non_zero = False
        for num in row:
            if num == 1:
                has_non_zero = True
                break  # 找到1就不用继续检查了
        if has_non_zero:
            x += 1
    return x

方法2：用内置函数简化 #

Python的any()函数可以直接判断迭代器里有没有True值，因为1在布尔判断里是True，0是False，所以一行就能判断是否非全0：

def xArrayCount(MX):
    x = 0
    for row in MX:
        if any(row):  # 只要行里有1，any()就返回True
            x += 1
    return x

方法3：一行式简洁写法 #

用生成器表达式配合sum()，一行搞定：

def xArrayCount(MX):
    return sum(1 for row in MX if any(row))

测试验证 #

拿你给的例子测试：

1. n=3的情况：

MX = [[0, 0, 0], [1, 0, 0], [0, 1, 0]]
print(xArrayCount(MX))  # 输出2，符合预期

2. n=4的情况：

MX = [[0, 0, 0, 0], [1, 0, 0, 0], [0, 1, 0, 0], [0, 0, 0, 0]]
print(xArrayCount(MX))  # 输出2，符合预期

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