基于首个>=1元素的二维数组修改：第三目标元素修改失败求助

你写的代码存在两个核心问题：

• 未定义变量c，运行时会直接抛出报错
• 嵌套的if逻辑完全偏离了统计>=1元素个数的思路，无法正确定位第三个目标元素

正确实现思路 #

要定位每个子列表中第N个>=1的元素，只需用一个计数器：遍历子列表时，每遇到一个符合条件的元素就递增计数器，当计数器等于目标序号（比如第三个就是3）时，修改该元素并跳出循环即可。

针对第三个>=1元素的代码 #

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

# 修改每个子列表中第三个>=1的元素，加100
for l in example_array:
    count = 0
    for i, x in enumerate(l):
        if x >= 1:
            count += 1
            if count == 3:
                l[i] += 100
                break

# 输出修改后的数组
for row in example_array:
    print(row)

运行后输出结果：

[0, 1, 101, 1, 0, 0, 0]
[2, 1, 101, 2, 2, 0, 0]
[0, 3, 1, 101, 3, 3, 0]
[4, 1, 101, 4, 0, 0, 0]
[0, 5, 1, 101, 5, 0, 0]
[0, 1, 1, 0, 0, 0, 0]  # 该子列表仅有2个>=1的元素，无修改

通用化封装（支持任意第N个元素） #

如果需要灵活修改第1、2、3...个符合条件的元素，可以封装成通用函数：

def modify_nth_valid_element(arr, target_n, add_val=100):
    for sub_list in arr:
        counter = 0
        for idx, val in enumerate(sub_list):
            if val >= 1:
                counter += 1
                if counter == target_n:
                    sub_list[idx] += add_val
                    break
    return arr

# 调用示例：修改每个子列表中第2个>=1的元素
modified_arr = modify_nth_valid_element(example_array, 2)

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