本文介绍如何将形状为 (m, n, 2) 的三维 numpy 数组中每个末维二元子数组（如 [1, 2]），依据字典映射快速转换为标量值，生成 (m, n) 形状的输出数组，并对比多种实现方式的性能与适用场景。
本文介绍如何将形状为 (m, n, 2) 的三维 numpy 数组中每个末维二元子数组（如 [1, 2]），依据字典映射快速转换为标量值，生成 (m, n) 形状的输出数组，并对比多种实现方式的性能与适用场景。
在科学计算与数据预处理中，常需将结构化坐标或索引对（如像素坐标、类别组合、键值对）批量映射为标量标签或权重。典型场景包括：图像特征编码、图神经网络中的边类型映射、或离散状态空间的查表赋值。给定一个 (m, n, 2) 的 NumPy 数组 input_array 和一个 Dict[Tuple[int, int], float] 类型的映射字典，目标是生成形状为 (m, n) 的 output_array，其中每个元素 output_array[i, j] = mapping[tuple(input_array[i, j])]。
最直观但低效的方式是嵌套循环或逐元素 np.where + np.all 掩码（如原始尝试），其时间复杂度随 m×n 增长且存在重复遍历。更优解是利用 NumPy 的向量化 reshape 与 Python 的生成器表达式协同完成“扁平化→查表→重塑”三步：
import numpy as np
def map_subarrays_fast(input_array, mapping):
"""
将 (m, n, 2) 数组中每个 [a, b] 子数组替换为 mapping[(a, b)]
返回 (m, n) 形状的浮点数组。
"""
# 展平前两维，得到 (m*n, 2)；逐行转为 tuple 并查字典；再重塑
flat_tuples = map(tuple, input_array.reshape(-1, 2))
mapped_values = [mapping[t] for t in flat_tuples]
return np.array(mapped_values, dtype=float).reshape(input_array.shape[:-1])
该方法简洁、可读性强，且在中小规模（如 m,n ≤ 200）下性能显著优于原始循环方案（实测快约 5–10 倍）。若输入中存在大量重复子数组（例如遥感影像中有限类别的波段组合），可进一步优化：先提取唯一子数组、批量查表、再通过 return_inverse 索引还原——这避免了重复字典查找，适合高重复率场景：
def map_subarrays_unique_optimized(input_array, mapping):
reshaped = input_array.reshape(-1, 2)
unique_vals, inverse_indices = np.unique(reshaped, axis=0, return_inverse=True)
# 对唯一值批量查表（注意：tuple 是不可哈希的 ndarray，需显式转换）
unique_keys = [tuple(row) for row in unique_vals]
unique_mapped = np.array([mapping[k] for k in unique_keys], dtype=float)
return unique_mapped[inverse_indices].reshape(input_array.shape[:-1])
⚠️ 注意事项：
- 字典键必须完全匹配子数组内容（顺序、类型、值）。例如 [1, 2] 与 [1.0, 2.0] 视为不同键，建议统一输入数据类型（如全 int 或全 float）。
- 若存在未覆盖的子数组（即 tuple(subarray) 不在 mapping 中），代码将抛出 KeyError。生产环境应添加健壮性检查，例如使用 mapping.get(tuple(sub), default_value)。
- np.fromiter 可替代 list 构造提升内存效率（尤其大数据集）：
out = np.fromiter(
(mapping[tuple(x)] for x in input_array.reshape(-1, 2)),
dtype=float,
count=input_array.size // 2
).reshape(input_array.shape[:-1])
总结：对于大多数场景，扁平化 + 列表推导式是最佳平衡点——代码简短、逻辑清晰、性能可靠；仅当子数组高度重复且规模极大时，才需引入 np.unique 预处理。避免过早优化，优先保证正确性与可维护性。
