本文介绍如何利用布尔索引替代显式循环，对三维 numpy 数组中满足 z=1 或 z=2 条件的整个切片（x,y,0:3）进行批量赋值，显著提升计算效率。
在处理多维 NumPy 数组（如形状为 (X, Y, Z) 的 3D 数组）时，若需根据某几个维度（例如 Z=1 和 Z=2）的值来统一修改对应位置的所有通道（包括 Z=0），使用 np.ndindex 或嵌套 for 循环虽逻辑直观，但性能极差——它强制 Python 层逐元素遍历，无法利用 NumPy 的向量化底层优化。
更高效的做法是：
构造布尔掩码（boolean mask）并执行向量化赋值
。核心思想是将条件判断“广播”到整个空间维度，生成一个形状为 (X, Y) 的二维布尔数组，再用它直接索引原始三维数组，实现整条 Z 轴切片的批量更新。
以下为优化后的代码示例：
✅ 关键说明：
Python 3.14.3
微软官方的 Python 扩展，是 VS Code 安装量最高的扩展（209M+）。集成 IntelliSense（通过 Pylance）、调试（通过 Python Debugger）、代码检查、格式化、重构和单元测试等功能。支持 Jupyter Notebook、虚拟环境管理和多 Python 版本切换。
下载
arr[..., 1] 等价于 arr[:, :, 1]，表示所有 (x, y) 位置上 Z=1 的切片；
| 是按元素逻辑或（注意不是 or），确保条件向量化；
arr[cond] = 0 利用了 NumPy 的高级索引机制：当布尔数组 cond 的形状与前导维度匹配时，NumPy 会自动将赋值广播至剩余维度（此处即 Z 轴全部三个通道）。
⚠️ 注意事项：
若仅需修改特定 Z 层（如只设 Z=1 和 Z=2 为 0），可显式指定：arr[cond, 1] = 0; arr[cond, 2] = 0；
确保条件表达式中各子项维度兼容（本例中 arr[...,1] 与 arr[...,2] 同为 (X,Y)，可直接逻辑运算）；
对超大数组，该方法内存友好且速度通常提升 10–100 倍以上，是 NumPy 最佳实践之一。
总结：摒弃 ndindex 循环，拥抱布尔索引——这是解锁 NumPy 高性能数据处理的核心习惯。
import numpy as np
# 示例：创建一个 (4, 5, 3) 的随机三维数组
arr = np.random.randint(0, 100, size=(4, 5, 3))
# 原始低效写法（不推荐）
# for i, j in np.ndindex(4, 5):
# if arr[i, j, 1] > 80 or arr[i, j, 2] < 22:
# arr[i, j, :] = 0 # 注意：这里应设全部 Z 维度
# ✅ 高效向量化写法
cond = (arr[..., 1] > 80) | (arr[..., 2] < 22) # 形状为 (4, 5)
arr[cond] = 0 # 自动沿最后一维广播：等价于 arr[cond, :] = 0