文章/答案/技术大牛

发布

问基于多个条目的行筛选
EN

Stack Overflow用户

提问于 2016-01-04 11:38:34

回答 1查看 37关注 0票数 0

在Python2-7中使用熊猫，我有以下数据(dfmain)：

    Col1    Letter
0   One     A
1   Two     B
2   Three   C
3   Four    A
4   Five    D
5   Six     B
6   Seven   C

我想过滤掉所有行，用'C‘或'D’作为字母，或者只有行'A‘和'B’。以下代码只适用于一个字母：

dftemp = dfmain[dfmain['Letter'] == 'A']

或者另一种选择：

dftemp = dfmain[dfmain['Letter'] != 'D']

然而，我如何在不止一封信中做到这一点，例如，我已经尝试过：

dftemp = dfmain[dfmain['Letter'] == 'A', 'B']

或者：

dftemp = dfmain[dfmain['Letter'] == ['A','B']]

还有一些其他的方法但是这些都会产生错误。有什么想法吗？

回答已采纳

发布于 2016-01-04 11:40:13

使用isin测试多个值的成员资格：

In [10]:
dftemp = df[df['Letter'].isin(['A', 'B'])]
dftemp

Out[10]:
   Col1 Letter
0   One      A
1   Two      B
3  Four      A
5   Six      B

票数 1

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/34590297

复制

相似问题

问基于多个条目的行筛选EN