计算每行的零数,并删除超过n个零的行

def count_zeros(row):
    """计算一行中零的数量"""
    count = 0
    for item in row:
        if item == 0:
            count += 1
    return count

def remove_rows(matrix, n):
    """删除超过n个零的行"""
    new_matrix = []
    for row in matrix:
        if count_zeros(row) <= n:
            new_matrix.append(row)
    return new_matrix

matrix = [
    [1, 0, 2, 0],
    [3, 0, 0, 4],
    [0, 0, 0, 0],
    [5, 6, 0, 7]
]

new_matrix = remove_rows(matrix, 2)
print(new_matrix)

[[1, 0, 2, 0], [5, 6, 0, 7]]