为什么pandas.Dataframe.drop()返回None?

时间:2016-08-08 10:52:39

标签: python pandas dataframe

在我的代码中,我从CSV中读取数据:

data = pandas.read_csv('dataset/job_functions.csv', names=["job","category"] ,skiprows=1).dropna().reindex()
num_jobs = data["job"].size

然后我想删除'category'标签不等于i的行:

data = data.drop(data[data.category!=i].index,inplace = True)
print(data.head())

甚至按索引列表返回无:

data = data.drop(data.index[[1,2,3]],inplace = True)

错误讯息:

Traceback (most recent call last):
File "sample.py", line 162, in 
  delete_common_words(27)
File "sample.py", line 92, in delete_common_words
  print(data.head())
AttributeError: 'NoneType' object has no attribute 'head'

以下是我使用drop()之前的数据:

                                                 job  category
0  офис   менеджер реализация гербицидовоформлени...         2
1  менеджер   отдел продажа работа с существующий...        27
2  ведущий   бухгалтер работа с вендер и поставщи...         1
3  менеджер   по продажа и продвижение продукт ус...        27
4  юрист   проведение юридический экспертиза прое...        13

1 个答案:

答案 0 :(得分:1)

看起来需要boolean indexing

import pandas as pd

data = pd.DataFrame({'category':['a','b', 'c']})
print (data)
  category
0        a
1        b
2        c

i = 'a'
print (data[data.category != i])
  category
1        b
2        c

print (data[~data.category.isin(['b','c'])])
  category
0        a

正如EdChum所述,如果使用inplace=True它会返回None,那么您可以使用:

#omit inplace=True
data = data.drop(data[data.category!=i].index)

或者:

#remove assigning
data.drop(data[data.category!=i].index,inplace = True)