未知索引下用循环与条件语句修改二维数组指定元素

示例数组 #

example_array = [[ 0, 1, 1, 0, 0, 0, 0], 
                 [ 2, 1, 1, 2, 0, 0, 0], 
                 [ 0, 3, 1, 1, 3, 0, 0], 
                 [ 4, 1, 1, 4, 0, 0, 0], 
                 [ 0, 5, 1, 1, 5, 0, 0], 
                 [ 0, 1, 1, 0, 0, 0, 0]]

需求说明 #

需要将每个子列表中第二个大于1的元素修改为「原值+100」。

现有可行代码（修改第一个符合条件的元素） #

以下代码可以正确修改每个子列表中第一个大于1的元素：

for l in example_array:
    for i, x in enumerate(l):
        if x > 1:
            l[i] += 100  # 注：原代码写的是-100，和预期结果不符，这里修正为+100
            break

错误尝试及问题 #

你尝试用嵌套if判断定位第二个符合条件的元素，但这段代码无法生效：

for l in example_array:
    for i, x in enumerate(l):
        if x > 1:
            if x > 1:
                l[i] -= 100
                break 

问题在于：嵌套的if x>1并没有起到计数作用，只要遇到第一个大于1的元素就会触发修改并break，根本不会处理第二个元素。

预期结果数组 #

修改后的目标数组如下：

example_array = [[ 0,  1, 101, 0, 0, 0, 0], 
                 [ 2, 101, 1,  2, 0, 0, 0], 
                 [ 0,  3, 101, 1, 3, 0, 0], 
                 [ 4, 101, 1,  4, 0, 0, 0], 
                 [ 0,  5, 101, 1, 5, 0, 0], 
                 [ 0,  1, 101, 0, 0, 0, 0]]

正确实现方法 #

方法一：计数器法（推荐） #

通过计数器记录找到的符合条件的元素数量，当计数到2时修改元素并停止遍历：

for l in example_array:
    match_count = 0
    for i, x in enumerate(l):
        if x > 1:
            match_count += 1
            # 找到第二个符合条件的元素时修改并跳出循环
            if match_count == 2:
                l[i] += 100
                break

方法二：索引收集法 #

先收集子列表中所有大于1的元素的索引，再修改第二个（确保存在的情况下）：

for l in example_array:
    # 筛选出所有大于1的元素的索引
    valid_indices = [idx for idx, val in enumerate(l) if val > 1]
    # 确保至少有2个符合条件的元素
    if len(valid_indices) >= 2:
        l[valid_indices[1]] += 100

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