pandas中单元素行索引与切片返回格式不一致如何解决

问题本质 #

pandas的.loc、.iloc索引存在默认降维逻辑：当传入单个标量值作为行索引时，返回结果会从二维DataFrame压缩为一维Series，你看到的纵向键值对展示样式就是Series的默认打印格式。此时调用len(结果.index)统计到的是Series的元素总数（即原表列数），和预期的行数统计逻辑不匹配。

可用解决方案 #

• 传列表包裹目标索引值（最推荐，兼容位置索引、标签索引所有场景）
  只要给.iloc/.loc传入的行索引参数是列表类型，pandas就不会触发降维，返回结果和切片得到的二维表结构完全一致：

  # 位置索引取第10行，返回DataFrame
  blast_output_scored.iloc[[10], :]
  
  # 标签索引取ID为IDgene.1的行，返回DataFrame
  blast_output_scored.loc[["IDgene.1"]]
  

  上述写法返回的结果做行数校验len(结果.index)会正确返回1，列结构、行索引都和原表完全对齐，不管是取单行还是后续扩展取多行，写法逻辑保持一致，不需要特殊判断。
• 一维结果转置还原为DataFrame
  如果已经拿到了Series类型的单行结果，可以通过转置快速还原为二维表格式：

  single_row_series = blast_output_scored.loc["IDgene.1"]
  single_row_df = single_row_series.to_frame().T
  

  转换后得到的同样是标准单行DataFrame，数据、索引信息不会丢失。

注意：不建议依赖[n:n+1]的切片写法固定取单行，尤其是标签索引场景下，如果存在重复行标签，切片范围很容易出现非预期结果。

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