条件选择数据框列

时间:2017-03-25 17:21:18

标签: python pandas

您好我想知道pandas列上的条件选择是如何工作的。在下面的代码中

In [162]: euro16
Out[162]: {'Goals': [16, 8], 'Team': ['Germany', 'England']}

In [163]: euro16_df = pd.DataFrame(euro16)

In [164]: euro16_df[euro16_df.Team == 'Germany']
Out[164]:
   Goals     Team
0     16  Germany

然而,当您尝试对涉及字符串访问的团队进行条件化时,即:假设所有团队都以' G'开头。我得到了一个KeyError。我非常感谢有关这里可能发生的事情的任何信息。

euro16_df[euro16_df.Team[0] == 'G']

2 个答案:

答案 0 :(得分:3)

使用str字符串访问者

euro16_df[euro16_df.Team.str[0] == 'G']

答案 1 :(得分:1)

也是str startswith。

euro16_df[euro16_df.Team.str.startswith('G')]