我现在正在学习数据帧。我一直纠结于如何获取带有标签索引的数据帧或表的子集。我知道这是一个非常简单的问题,但我在熊猫文档中找不到解决方案。希望有人能帮助我。谢谢你的帮助。
因此,我有一个名为df_teams
的数据帧,如下所示:在此处输入图像描述
如果我想获得特定团队的子表'Warriors',我可以使用df_teams[df_teams['nickname']=='Warriors']
,生成数据帧形式的行。我的问题是,如果我想得到更多球队的子表,比如说我想要"勇士"one_answers"老鹰"的信息来组成一个新的表,该怎么办?我可以通过使用逻辑索引并在一行代码中完成类似的操作吗?
您可以使用'|'字符对这两个条件执行逐位或。
df_teams[(df_teams['nickname']=='Warriors')|(df_teams['nickname']=='Hawks')]
或者,如果您有一个要检查的值列表,则可以使用isin方法返回具有列表中某个值的行。
E.g
df_teams[df_teams['nickname'].isin(['Warriors','Hawks'])]