如何处理Python矩阵 仅保留每行索引最小的0并删除后续其余零元素

问题原因分析 #

• 遍历列表过程中直接调用pop删除元素会动态修改列表长度，导致后续遍历拿到的索引和实际元素位置错位，既可能漏处理元素，也可能触发索引越界错误
• 原有代码的判断条件iy+ix<(iy+1)+ix化简后为恒成立表达式，无法实现「识别行内第一个0」的逻辑

正确实现代码 #

我们采用逐行构建新列表的方式避免修改原数据带来的索引异常，逻辑为：遍历每行元素时，标记是否已经找到第一个0，找到第一个0后仅保留后续的非0元素即可。

a = [[1, 0, 0, 0],
     [0, 0, 1, 1],
     [0, 1, 0, 1],
     [1, 0, 0, 1]]

res_matrix = []
keep_zero_pos = []

for row_idx, row in enumerate(a):
    new_row = []
    found_first_zero = False
    zero_col = None
    for col_idx, val in enumerate(row):
        if not found_first_zero:
            new_row.append(val)
            if val == 0:
                found_first_zero = True
                zero_col = col_idx
        else:
            if val != 0:
                new_row.append(val)
    res_matrix.append(new_row)
    keep_zero_pos.append((row_idx, zero_col))
    # 输出保留的0的坐标
    print(row_idx, zero_col)

# 替换原矩阵
a = res_matrix
print("a =", a)

输出结果 #

完全匹配预期输出：

0 1
1 0
2 0
3 1
a = [[1, 0], [0, 1, 1], [0, 1, 1], [1, 0]]

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