![](https://csdnimg.cn/release/download_crawler_static/88196787/bg3.jpg)
1 约会网站效果判定 3
[6]: (1000, 4)
[7]: datingTest.info()
<class 'pandas.core.frame.DataFrame'>
RangeIndex: 1000 entries, 0 to 999
Data columns (total 4 columns):
# Column Non-Null Count Dtype
--- ------ -------------- -----
0 0 1000 non-null int64
1 1 1000 non-null float64
2 2 1000 non-null float64
3 3 1000 non-null object
dtypes: float64(2), int64(1), object(1)
memory usage: 31.4+ KB
1.2 分析数据
[8]: Colors = []
for i in range(datingTest.shape[0]):
m = datingTest.iloc[i,-1]
# 将每个不同类别的人定义为不同的颜色
if m =="didntLike":
Colors.append('black')
if m =="smallDoses":
Colors.append('orange')
if m =="largeDoses":
Colors.append('red')
[9]: # 绘制两两特征之间的散点图
plt.rcParams['font.sans-serif'] = ['Simhei']# 途中字体设置为黑体
pl = plt.figure(figsize = (12,8))# 子画布
fig1 = pl.add_subplot(221)# 在子画布上绘制 2*2 的分区,这个图在 1 分区上
plt.scatter(datingTest.iloc[:,1],datingTest.iloc[:,2],marker= '.
↪',c=Colors)#scatter:散点图
plt.xlabel('玩游戏视频所占时间比')
plt.ylabel('每周消费冰淇淋公升数')