如何在Python中按指定行列将零矩阵元素置为1？

现有一个12×18的零矩阵zero，以及列表newArr2——其中每个元素对应矩阵某一行需要置1的列索引（如果元素为0，则该行不做修改）。例如第一行需要将第11、17列的0改为1，第二行无修改。尝试以下代码未解决问题：

for j in range (12):
    for i in range (len(newArr2[j])):
        rr=zero[j][newArr2[j][i]]=1

问题分析 #

原代码存在两个核心问题：

1. 无效操作触发报错：当newArr2[j]为0时，调用len(newArr2[j])会直接报错，因为0不是可迭代对象，无法获取长度。
2. 冗余且无意义的赋值：rr=zero[j][newArr2[j][i]]=1这行代码里的rr=完全多余，同时未对列索引的有效性做检查，若索引超出0-17范围会直接抛出索引错误。

修正后的代码 #

# 遍历矩阵的每一行
for row_idx in range(12):
    cols_to_set = newArr2[row_idx]
    # 仅对需要修改的行执行操作
    if cols_to_set != 0:
        for col_idx in cols_to_set:
            # 确保列索引在有效范围内（0到17）
            if 0 <= col_idx < 18:
                zero[row_idx][col_idx] = 1

代码说明 #

• 先判断当前行是否需要修改，跳过值为0的行，避免无效操作报错。
• 增加列索引范围校验，防止因非法索引导致程序崩溃。
• 移除冗余赋值，简化代码逻辑。

如果newArr2中不需要修改的行对应的元素是空列表而非0，可使用更简洁的版本：

for row_idx in range(12):
    for col_idx in newArr2[row_idx]:
        if 0 <= col_idx < 18:
            zero[row_idx][col_idx] = 1

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