如何使用enumerate()修改二维数组中每个子列表的首个大于1的元素
问题分析与解决
先拆解你原代码里的几个核心错误:
- 内层循环用了
sub_list[1:][1:1]切片,[1:1]会返回空列表,直接导致内层循环根本没执行,自然不会修改任何元素。 - 条件判断
entry > 1 and j != 1 and j != len(sub_list)完全偏离需求,我们要找的是每个子列表第一个大于1的元素,不需要排除这些位置。 - 没有在找到目标元素后终止内层循环,可能会误改后续符合条件的元素(即使你需求只改第一个)。
正确代码实现
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]] for i, sub_list in enumerate(example_array): for j, entry in enumerate(sub_list): if entry > 1: example_array[i][j] += 100 break # 找到第一个符合条件的元素就终止内层循环,确保只修改第一个 print(example_array)
运行结果
输出完全匹配你期望的正确数组:
[[0, 1, 1, 0, 0, 0, 0], [102, 1, 1, 2, 0, 0, 0], [0, 103, 1, 1, 3, 0, 0], [104, 1, 1, 4, 0, 0, 0], [0, 105, 1, 1, 5, 0, 0], [0, 1, 1, 0, 0, 0, 0]]
额外提示
如果需要保留原数组不被修改,可以先做深拷贝再操作:
import copy new_array = copy.deepcopy(example_array) for i, sub_list in enumerate(new_array): for j, entry in enumerate(sub_list): if entry > 1: new_array[i][j] += 100 break
内容的提问来源于stack exchange,提问作者Patstro
相关产品推荐
相关产品推荐

