如何对三维numpy ndarray的最后一维元素执行特定基本数学运算

直接利用numpy的向量化操作就能高效完成这个计算，不需要循环遍历每个元素：

import numpy as np

# 假设你的data已经是(984, 1977, 2)的ndarray
band1 = data[..., 0]  # 提取第一个波段，维度(984, 1977)
band2 = data[..., 1]  # 提取第二个波段，维度(984, 1977)

# 执行公式计算
result = (band1 - band2) / (band1 + band2)

关键说明： #

• data[..., 0] 是numpy的索引语法，...代表取所有前面的维度（这里就是前两个维度984和1977），最后取第0个元素，等价于data[:, :, 0]，写法更简洁。
• numpy会自动对同维度数组执行元素级运算，这比手动写for循环效率高得多，尤其适配大尺寸数组场景。
• 如果存在band1 + band2 = 0的情况，会触发除零警告，可以通过添加极小值规避：

  epsilon = 1e-10
  result = (band1 - band2) / (band1 + band2 + epsilon)
  

如果需要结果保持三维维度（比如(984,1977,1)），可以用np.expand_dims扩展：

result_3d = np.expand_dims(result, axis=-1)

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