基于多个条件得到df的分段

2024-10-04 11:26:34 发布

您现在位置:Python中文网/ 问答频道 /正文

我试图根据多个条件从df中提取行,在选择任何行之前必须满足所有条件,否则什么都没有。在

我的df

columns = ['is_net', 'is_pct', 'is_mean', 'is_wgted', 'is_sum']
index = ['a','b','c','d']
data = [['True','True','False','False', 'False'],
       ['True','True','True','False', 'False'],
       ['True','True','False','False', 'True'],
       ['True','True','False','True', 'False']]

df = pd.DataFrame(columns=columns, index=index, data=data)
df

    is_net  is_pct  is_mean is_wgted    is_sum
a   True    True    False   False   False
b   True    True    True    False   False
c   True    True    False   False   True
d   True    True    False   True    False

我的条件必须采用此格式:

^{pr2}$

预期产量:

    is_net  is_pct  is_mean is_wgted    is_sum
d   True    True    False   True    False

Tags: columnsfalsetruedataframedfdataindexnet
1条回答
网友
1楼 · 发布于 2024-10-04 11:26:34

我认为诀窍是把conditions变成一个系列:

>>> pd.Series(conditions)
is_mean     False
is_net       True
is_pct       True
is_sum      False
is_wgted     True
dtype: object
>>> (df == pd.Series(conditions))
  is_mean is_net is_pct is_sum is_wgted
a    True   True   True   True    False
b   False   True   True   True    False
c    True   True   True  False    False
d    True   True   True   True     True
>>> (df == pd.Series(conditions)).all(axis=1)
a    False
b    False
c    False
d     True
dtype: bool
>>> df[(df == pd.Series(conditions)).all(axis=1)]
  is_net is_pct is_mean is_wgted is_sum
d   True   True   False     True  False

相关问题 更多 >