选择等于pandas数据帧中的字段的列

时间:2013-12-18 06:26:25

标签: python pandas dataframe

我的Pandas DataFrame看起来像这样:

0                   STUN
1                  Webex
2                    PPP
3                MyVideo
4                Icecast
5               PPSTREAM
6                    FTP
7                   SPDY
8     Thunder/Webthunder
9                    IRC
10            CitrixGoTo
11                 FLASH
12               GameKit
13                   RDP
14                IMplus
...
505         unknown
506      BitTorrent
507          ISAKMP
508            HTTP
509       REALMEDIA
510     Silverlight

由此我必须选择等于HTTP和SSH的列。

1 个答案:

答案 0 :(得分:3)

您可以在列上使用isin Series方法:

df[df[column_name].isin(['HTTP', 'SSH'])]

另一种方法是检查是否相等(最有可能的是更快):

df[(df[column_name] == 'HTTP') | (df[column_name] == 'SSH'])]