使用iloc[0]提取DataFrame行时如何自动校验仅1行且无需显式判断长度

方案1：使用pandas原生one()方法（pandas≥2.2.0 推荐） #

pandas 2.2.0及以上版本已经内置了符合你需求的原生方法，不需要自己写判断逻辑：

• 调用df.one(axis='index')会自动校验DataFrame恰好只有1行，符合条件时返回对应行的Series，行数不等于1时主动抛出ValueError
• 调用df.one(axis='columns')会自动校验恰好只有1列，符合条件返回对应列的Series
• 无参调用会校验行列数均为1，符合条件返回标量

示例代码：

# 仅校验行数为1，返回对应行Series
s = df.one(axis='index')

方案2：低版本pandas兼容方案 #

如果你的pandas版本低于2.2.0，没有内置的one()方法，可以通过以下两种方式处理：

方式A：封装公用工具函数 #

只需要写一次判断逻辑，后续所有场景直接调用函数即可，不会产生冗余代码：

import pandas as pd

def get_single_row(df: pd.DataFrame) -> pd.Series:
    if len(df) != 1:
        raise ValueError(f"预期DataFrame仅有1行，实际为{len(df)}行")
    return df.iloc[0]

后续使用时直接调用即可：

s = get_single_row(df)

方式B：使用squeeze()方法 #

如果不需要主动抛错，只需要在恰好1行时返回Series，可以用squeeze()方法：

# 仅在行数为1时返回Series，行数不为1时返回原DataFrame
s = df.squeeze(axis=0)

如果需要强制校验，可以额外加一句断言，代码量也非常少：

s = df.squeeze(axis=0)
assert isinstance(s, pd.Series), "DataFrame行数不等于1"

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