本文介绍在 Pandas 透视表（pivoted DataFrame）中高效获取非 NaN 值的方法，核心是利用 .stack() 将二维稀疏结构转为紧凑的 Series，支持通过元组索引直接、安全地访问有效值，避免手动处理 KeyError 或冗余的双向查找逻辑。
本文介绍在 pandas 透视表（pivoted dataframe）中高效获取非 nan 值的方法，核心是利用 `.stack()` 将二维稀疏结构转为紧凑的 `series`，支持通过元组索引直接、安全地访问有效值，避免手动处理 keyerror 或冗余的双向查找逻辑。
当对距离等成对计算结果进行透视（如 df.pivot(index="id1", columns="id2", values="distance")）后，得到的 DataFrame 通常稀疏——大量单元格为 NaN，尤其在非对称或不完全配对场景下。此时若需在循环中按动态组合（如 ("id_A", "id_B")）查询值，直接使用 piv.loc[id1, id2] 易触发 KeyError 或返回 NaN，而逐个 try/except 切换行列顺序（如 piv.loc[id2, id1]）不仅代码冗余，还违背透视表的设计初衷。
更优雅的解法是：将透视表“压平”为非空值驱动的结构。DataFrame.stack() 方法正是为此设计——它会自动丢弃所有 NaN，返回一个以 (row_label, column_label) 为 MultiIndex、以原值为数据的 Series：
# 假设 piv 是已生成的透视表
non_nan_series = piv.stack(future_stack=False) # pandas ≥ 2.1 可省略 future_stack=False
# 直接通过元组索引获取非空值（安全、高效）
value = non_nan_series.loc[("gamma", "c")] # 返回标量值，非 NaN
# 批量随机采样有效组合
import random
valid_pairs = random.sample(non_nan_series.index.tolist(), k=3)
for idx in valid_pairs:
print(f"{idx} → {non_nan_series.loc[idx]}")该 Series 的索引天然支持 .loc 元组查询，且仅包含原始数据中存在的有效组合，从根本上规避了 NaN 和 KeyError。相比保留完整透视表再做条件判断，此方法内存更优、查询更快，也更符合函数式数据流思想。
⚠️ 注意事项：
- 若原始数据存在重复索引（如相同 id1 和 id2 多次出现），pivot() 会报错；建议先用 pivot_table(..., aggfunc="first") 或预聚合。
- stack() 默认保留原始列名层级，若需扁平化列名，可链式调用 .droplevel(0)（当仅有一层列时）。
- 对于需频繁双向查询（id1→id2 或 id2→id1）的对称距离矩阵，更推荐在构建阶段就确保对称性（如 pd.concat([df, df.rename(columns={"id1":"id2","id2":"id1"})])），再统一 pivot，而非运行时兜底。
综上，.stack() 不仅是技术捷径，更是数据形态与访问模式匹配的设计范式：让结构服务于查询，而非让查询迁就结构。
