python - 访问 pandas 数据框中单行的推荐方法？

Question

我经常想使用多个条件选择 Pandas DataFrame 的单行。

为了进一步处理，将其作为系列或字典访问是有用的，而不是作为 1xW DataFrame 访问。

我可以这样做:

r = df.loc[ (df['col1'] == v1) & (df['col2'] == v2) ]
if len(r.index) != 1:
  raise ValueError(...)
r = r.iloc[0]

这感觉有点笨拙。有更好的推荐方法吗？

Answer 1

所有内置访问器都不会执行您想要的操作。您可以构建自己的访问器并对其进行猴子修补:

def get(df, **kwargs):
    cond = [True] * df.shape[0]
    for col, value in kwargs.items():
        cond &= df[col] == value

    result = df[cond]
    if result.shape[0] > 1:
        raise ValueError('Key must be unique')
    return result.iloc[0]

pd.DataFrame.get = get

# Usage
df.get(col1=..., col2=..., col3=...)

但是，这相对较慢，因此不要在紧密循环中调用它。