问题背景

使用matplotlib将DBSCAN分类结果散点图可视化时提示此TypeError

源代码:
from sklearn.cluster import DBSCAN import pandas as pd import
matplotlib.pyplot as plt def devicesDbscan():
path="./unique_deviceID_lng_lat.csv"
df=pd.read_csv(path,header=None,names=["DEVICE_ID","LNG","LAT"])
print(df.shape()) X_df=df.drop("DEVICE_ID",axis=1) #去掉ID
y_pred=DBSCAN(eps=0.1,min_samples=1,n_jobs=20).fit_predict(X_df) #拟合并返回预测标签
count_clusters = len(set(y_pred)) #聚类簇的数目
print("clusters的数目:"+str(count_clusters))
plt.scatter(X_df[:,0],X_df[:,1],c=y_pred) #-->提示错误地方 plt.show() if
__name__=="__main__": devicesDbscan()
解决

将DataFrame对象X_df转成ndarray数组即可
from sklearn.cluster import DBSCAN import pandas as pd import
matplotlib.pyplot as plt def devicesDbscan():
path="./unique_deviceID_lng_lat.csv"
df=pd.read_csv(path,header=None,names=["DEVICE_ID","LNG","LAT"])
print(df.shape()) X_df=df.drop("DEVICE_ID",axis=1)
y_pred=DBSCAN(eps=0.1,min_samples=1,n_jobs=20).fit_predict(X_df) count_clusters
= len(set(y_pred)) print("clusters的数目:"+str(count_clusters))
plt.scatter(X_df.values[:,0],X_df.values[:,1],c=y_pred) plt.show() if
__name__=="__main__": devicesDbscan()
 

 

技术
©2019-2020 Toolsou All rights reserved,
iPhone 12售价、配置齐曝光:砍掉64GB、电池2227mAh起步el-select获取选中项label值最优化方法总结:公式解、数值优化、求解思想Thread.getContextClassLoader与Thread.getClassLoader()区别(精华2020年6月2日更新) TypeScript函数详解css基础2:flex 多行均分有间隙布局【Python】读取txt文件,获取指定行中指定位置数据Vue el-select 获取label值(精华)2020年7月13日 微信小程序 页面间通信Dijkstra算法的Python实现-最短路径问题