import pandas as pd from sklearn.cluster import DBSCAN import matplotlib.pyplot as plt from sklearn.preprocessing import MinMaxScaler # 选择需要聚类的三列数据并转换为numpy数据 data = df[['discounted_price','discount_percentage','rating_count']].values # 创建一个MinMaxScaler对象 scaler = MinMaxScaler() # 对DataFrame进行特征缩放 data = scaler.fit_transform(data) # 找出所有特征值不在范围内的行 outliers = df.loc[(data<0)|(data>0.6)].dropna(how='all').index # 删除这些行 df = df.drop(outliers) # 选择需要聚类的三列数据 data = df[['discounted_price','discount_percentage','rating_count']].values # 创建一个MinMaxScaler对象 scaler = MinMaxScaler() # 对data进行特征缩放 data = scaler.fit_transform(data) ## 构建DBSCAN聚类模型 dbscan = DBSCAN(eps=0.1, min_samples=3) dbscan.fit(data) # 获取聚类结果 labels = dbscan.labels_ ## 可视化聚类结果 fig = plt.figure() ax = fig.add_subplot(projection='3d') ax.scatter(data[:,0], data[:,1], data[:,2], c=labels) ax.set_xlim([0,0.6]) ax.set_ylim([0,0.6]) ax.set_zlim([0,0.6]) # 添加x轴和y轴标签 ax.set_xlabel('discounted_price') ax.set_ylabel('discount_percentage') ax.set_zlabel('rating_count') plt.show()怎样能让他实现在窗口交互旋转
时间: 2024-01-14 22:04:08 浏览: 87
你可以使用`mpl_toolkits.mplot3d`提供的`Axes3D`来创建一个3D坐标轴对象,并在创建坐标轴对象时设置`proj_type='interactive'`参数,来实现在窗口交互旋转的效果。具体代码如下:
```python
import pandas as pd
from sklearn.cluster import DBSCAN
import matplotlib.pyplot as plt
from sklearn.preprocessing import MinMaxScaler
from mpl_toolkits.mplot3d import Axes3D # 导入3D坐标轴对象
# 选择需要聚类的三列数据并转换为numpy数据
data = df[['discounted_price','discount_percentage','rating_count']].values
# 创建一个MinMaxScaler对象
scaler = MinMaxScaler()
# 对DataFrame进行特征缩放
data = scaler.fit_transform(data)
# 找出所有特征值不在范围内的行
outliers = df.loc[(data<0)|(data>0.6)].dropna(how='all').index
# 删除这些行
df = df.drop(outliers)
# 选择需要聚类的三列数据
data = df[['discounted_price','discount_percentage','rating_count']].values
# 创建一个MinMaxScaler对象
scaler = MinMaxScaler()
# 对data进行特征缩放
data = scaler.fit_transform(data)
## 构建DBSCAN聚类模型
dbscan = DBSCAN(eps=0.1, min_samples=3)
dbscan.fit(data)
# 获取聚类结果
labels = dbscan.labels_
## 可视化聚类结果
fig = plt.figure()
# 创建3D坐标轴对象,并设置proj_type参数为'interactive'
ax = fig.add_subplot(111, projection='3d', proj_type='interactive')
ax.scatter(data[:,0], data[:,1], data[:,2], c=labels)
ax.set_xlim([0,0.6])
ax.set_ylim([0,0.6])
ax.set_zlim([0,0.6])
# 添加x轴和y轴标签
ax.set_xlabel('discounted_price')
ax.set_ylabel('discount_percentage')
ax.set_zlabel('rating_count')
plt.show()
```
这样,你就可以在运行代码后通过鼠标在窗口中拖拽来旋转3D坐标轴了。
阅读全文