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

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

我正在尝试确定此 Pandas Dataframe 的适当语法，以了解如何为“Channel”列等于 A 或 B 的所有实例编制索引。找到所有实例后，我想将它们打印出来.此外，我希望能够调用每个索引以在脚本中进一步应用。

我希望显示为：

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

然后我想要一个“for 循环”，它通过并分别打印出每个索引实例，以便轻松识别和单独调用它们以在脚本中进一步使用。有人可以请教吗？

Answer 1

使用query

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