如何从两列中获取包含特定值的熊猫数据框的行索引？

Question

如何使用两个列条件获取行索引？

例如，如果我有一个看起来像这样的数据框：

Dataframe

如果我指定要让“口袋”为0且“牛仔布”为1的行，我希望返回整数“ 1”，因为1是满足这两个列条件的行索引。

Answer 1

我认为需要按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