如何逐个遍历numpy二维数组元素并按条件输出网格结果？

问题分析 #

你遇到的ValueError是因为循环对象错误：如果直接遍历整个数组M，每次迭代的row是整行的numpy数组，此时row ==1会返回一个布尔数组，if语句无法直接判断数组的布尔值，导致歧义错误。同时你原代码仅遍历了第一行M[0]，无法处理整个二维数组。

方法一：嵌套循环逐个遍历元素 #

通过两层循环分别遍历行和每行内的元素，逐个判断并生成输出：

import numpy as np
M = np.array([[1,1,5], [1,1,4], [1,1,1]])

for row in M:
    output_line = []
    for elem in row:
        output_line.append('2' if elem == 1 else '3')
    print(' '.join(output_line))

运行后会输出：

2 2 3
2 2 3
2 2 2

方法二：Numpy向量化操作（推荐） #

Numpy原生支持向量化运算，无需循环即可高效完成值替换，尤其适合大数组：

import numpy as np
M = np.array([[1,1,5], [1,1,4], [1,1,1]])

# 用np.where直接生成替换后的数组
result = np.where(M == 1, 2, 3)

# 按行打印结果
for row in result:
    print(' '.join(map(str, row)))

或者分步实现逻辑：

import numpy as np
M = np.array([[1,1,5], [1,1,4], [1,1,1]])

# 创建全为3的数组，再将等于1的位置替换为2
result = np.full(M.shape, 3)
result[M == 1] = 2

for row in result:
    print(' '.join(map(str, row)))

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