Python：获取与列值匹配的所有行的索引[duplicate]

Question

我有以下数据框：

Rec  Channel  Value1  Value2 
Pre             10      20
Pre             35      42
Event    A      23      39
FF              50      75
Post     A      79      11
Post     B      88      69

我正在尝试为这个熊猫数据框确定适当的语法，如何为列“通道”等于A或B的所有实例编制索引。一旦找到所有实例，我想打印出来。此外，我希望能够在脚本中调用每个索引以获得进一步的应用程序。

我希望显示器是：

Rec  Channel  Value1  Value2
Event   A       23      39
Post    A       79      11
Post    B       88      69

然后我想有一个“for循环”，它遍历并分别打印出每个索引实例，这样就很容易识别并单独调用它们，以便在脚本中进一步使用。有人能给点建议吗？

Answer 1

使用查询

res = df.query('Channel in ["A", "B"]')

res

     Rec Channel  Value1  Value2
2  Event       A      23      39
4   Post       A      79      11
5   Post       B      88      69

如果您需要索引以便用于其他目的：

要么

idx = res.index

或者，不做res

idx = df.index[df.Channel.isin(['A', 'B'])]

Answer 2

您可以为此使用pd.Series.isin：

res = df[df['Channel'].isin({'A', 'B'})]

print(res)

#      Rec Channel  Value1  Value2
# 2  Event       A      23    39.0
# 4   Post       A      79    11.0
# 5   Post       B      88    69.0

按索引返回第二行：

res2 = res.loc[2]

print(res2)

# Rec        Event
# Channel        A
# Value1        23
# Value2        39
# Name: 2, dtype: object