我經常想使用多個條件選擇單行 Pandas DataFrame。對于進一步處理，將其作為 Series 或 Dict 訪問是有用的，而不是作為 1xW DataFrame。我可以這樣做：r = df.loc[ (df['col1'] == v1) & (df['col2'] == v2) ]if len(r.index) != 1: raise ValueError(...)r = r.iloc[0]這感覺有點笨拙。有沒有更好的推薦方法？
1 回答
森林海
TA貢獻2011條經驗 獲得超2個贊
沒有一個內置訪問器會做你想做的事。您可以構建自己的訪問器并對其進行猴子補丁：
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=...)
但是，這相對較慢，因此不要在緊密循環中調用它。
添加回答
舉報
0/150
提交
取消