如何使用两列条件获取行索引?
例如,如果我有一个如下所示的数据框:
如果我指定我想要“Pocket”为 0 且“Denim”为 1 的行,我想要返回整数“1”,因为 1 是满足这两个列条件的行索引。
我认为需要 filter by boolean indexing
,但对于next
返回第一个匹配的行索引的一般解决方案,如果不存在值(掩码返回所有False
值)使用默认参数:
df = pd.DataFrame({'Product':['Jeans','Shorts','Jeans'],
'Pocket':[1,0,1],
'Denim':[0,1,1]}, index=[4,8,10])
print (df)
Product Pocket Denim
4 Jeans 1 0
8 Shorts 0 1
10 Jeans 1 1
mask = (df["Pocket"] == 0) & (df["Denim"] == 1)
a = next(iter(df[mask].index), 'not exist')
print (a)
8
mask = (df["Pocket"] == 0) & (df["Denim"] == 0)
a = next(iter(df[mask].index), 'not exist')
print (a)
not exist
本文收集自互联网,转载请注明来源。
如有侵权,请联系 [email protected] 删除。
我来说两句