Python中统计矩阵全零行数的代码bug排查求助

Hey there! Let's get to the bottom of why your code isn't correctly counting rows where every element is 0. I’ve debugged similar issues plenty of times, so here are the most common pitfalls to check, plus a working example to compare against your code.

Common Bugs to Look For #

• Incorrect element check: A super common mistake is comparing an entire row directly to 0 (like if row == 0:). Since a row is a list, this will never evaluate to True—you need to verify every element in the row is 0 instead. Use all(element == 0 for element in row) for this check.
• Off-by-one loop errors: If you’re using range(len(matrix)) but accidentally subtract 1 (e.g., range(len(matrix)-1)), you’ll skip the last row entirely. Stick to iterating directly over the matrix rows (for row in matrix:) to avoid this.
• Accidental matrix modification: If your loop modifies elements in the matrix while checking them, you might alter the rows you’re supposed to count. Make sure your counting logic doesn’t change any values in the matrix.
• Ignoring non-integer zeros: If your matrix has floating-point values (like 0.0), element == 0 still works, but if you’re dealing with precision issues (e.g., 0.0000001 that should be 0), you might need a tolerance check like abs(element) < 1e-9.

Working Example Code #

Here’s a solid implementation that avoids all these issues:

def count_zero_rows(matrix):
    zero_row_count = 0
    for row in matrix:
        # Verify every element in the row is 0
        if all(element == 0 for element in row):
            zero_row_count += 1
    return zero_row_count

# Test with a sample matrix
sample_matrix = [
    [0, 0, 0],
    [1, 2, 0],
    [0, 0, 0],
    [3, 4, 5]
]
print(count_zero_rows(sample_matrix))  # Should output 2

If your code still isn’t working after checking these points, feel free to share your code snippet—I’ll help you spot exactly where things are going wrong.

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