数据集如下
storeid,revenue,profit,country101,11434,2345,IN101,12132,3445,US102,21343,4545,CH103,34423,3432,CH103,43435,3234,JP103,34345,3335,IN
代码如下
import pandas as pdfrom sklearn.cluster import DBSCANimport matplotlib.pyplot as pltimport numpy as npfrom sklearn.preprocessing import StandardScalerimport seaborn as snsfrom pylab import rcParamsfrom collections import Countersns.set(rc={'figure.figsize':(11.7,8.27)})sns.set_style('whitegrid')df = pd.read_csv('1.csv',index_col=None)df.head()df.columns = df.columns.str.replace(' ', '')dummies = pd.get_dummies(data = df)del dummies['Unnamed:0']model = DBSCAN(eps = 2.25, min_samples=19).fit(dummies)print (model)target = dummies.iloc[:,0]data = dummies.iloc[:,1:-1]outliers_df = pd.DataFrame(data)print (Counter(model.labels_))print(outliers_df(model.labels_==-1))
print(outliers_df(model.labels_==-1))
抛出 TypeError: ‘DataFrame’ 对象不可调用
回答:
使用 布尔索引
与 []
来按掩码过滤:
print(outliers_df[model.labels_==-1]) revenue profit country_CH country_IN country_JP0 11434 2345 0 1 01 12132 3445 0 0 02 21343 4545 1 0 03 34423 3432 1 0 04 43435 3234 0 0 15 34345 3335 0 1 0