使用 Pandas 数据框中的字数统计来删除仅包含一个单词的行

我有一个包含 2 条记录的数据框（数据）：

id    text
0001  The farmer plants grain
0002  tuna

我想统计一下里面的单词数text该数据框的列并删除只有一个单词的行。

我知道如何计算单词数：

count = data['text'].str.split().str.len()

如何使用结果运行 IF 语句来删除数据框中的行？任何 IF 语句，例如...

if count == 1:
    print('drop')

...导致此错误：

Traceback (most recent call last):

  File "<ipython-input-118-b3fcb0218e8e>", line 32, in <module>
    if count == 1:

  File "C:\Users\taca\AppData\Local\Continuum\Anaconda3\lib\site-packages\pandas\core\generic.py", line 917, in __nonzero__
    .format(self.__class__.__name__))

ValueError: The truth value of a Series is ambiguous. Use a.empty, a.bool(), a.item(), a.any() or a.all().

我已经阅读了 Pandas 文档和其他有关此错误的问题，但我似乎无法找到正确应用于 IF 语句问题的解决方案。

任何意见是极大的赞赏！由于我对 SO 比较陌生，请告诉我是否可以做任何事情来改进我的问题。

使用面膜：

dropped = data[~(count==1)].copy()

解释：

所以假设 df 有：

data = pd.DataFrame({'text': ['hello my name is','hey']})

使用您的 count 方法，您可以检查它是否=1，创建一个布尔掩码：

count = data['text'].str.split().str.len()
~(count==1)
Out[18]: 
0     True
1    False
Name: text, dtype: bool

现在你可以应用这个面具：

data[~(count==1)]
Out[22]: 
               text
0  hello my name is