NumPy数组修改求助：对小于50的元素运算后保留原形状

错误原因分析 #

你编写的循环代码存在两个关键问题：

1. array < 50会生成一个与原数组形状一致的布尔数组，直接将其放在if条件中时，Python无法判断整个布尔数组的"真值"（因为数组包含多个元素），因此抛出ValueError。
2. 循环中for i in array的i是数组的每一行，即使条件判断正确，操作的也是整行而非单个元素，不符合需求。

正确解决方案 #

NumPy的核心优势是向量化操作，完全不需要用Python循环处理单个元素，以下两种方法都能高效完成需求：

方法1：布尔掩码索引

通过布尔数组定位需要修改的元素，直接批量操作：

import numpy as np

# 生成原数组
array = np.arange(2, 201, 2).reshape(25, 4)
# 创建布尔掩码，标记所有小于50的元素
mask = array < 50
# 对掩码选中的元素执行 (元素+5)*2 的操作
array[mask] = (array[mask] + 5) * 2

方法2：np.where 函数

用np.where可以一行代码完成条件赋值，更简洁：

import numpy as np

array = np.arange(2, 201, 2).reshape(25, 4)
# 语法：np.where(条件, 满足条件时的操作, 不满足条件时的取值)
array = np.where(array < 50, (array + 5) * 2, array)

两种方法都会保持数组25×4的形状，且效率远高于Python循环，这也是NumPy处理数组的标准方式。

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