机器学习之绘图(二)

有这样一组数据 (消费 小费 性别 吸烟者或非吸烟者 日期 时间)

# 我们的任务:吸烟顾客与不吸烟顾客的消费金额与小费绘制散点图
#  话不多说 上代码
#  首先引入需要用到的模块
import matplotlib.pyplot as plt
#  读取数据
df = pd.read_excel("./tips.xlsx",'sheet1')

#  分别取出吸烟者和非吸烟者数据

#  有了这些数据 我们就可以绘制散点图
	plt.scatter(a,b,marker="*",label="非吸烟者")    #  marker 指定形状
	plt.scatter(c,d,marker="<",color="red",label="吸烟者")
	plt.xlabel("小费")
	plt.ylabel("消费")
	plt.legend() #设置图例
	plt.show()

效果图如下

吸烟者和不吸烟者 给的小费 一目了然

吸烟者随着消费的递增 给的小费也在递增 而非吸烟者虽然消费高 但给的小费最高只在6元左右 只有一例是消费了50元 给了10元小费

还有另外一种展示效果

import matplotlib.pyplot as plt
df = pd.read_excel("./tips.xlsx",'sheet1')

# 取出非吸烟者数据
smoker = df[df["smoker"]==  "No"]
# 取出吸烟者数据
smoker1 = df[df["smoker"]=="Yes"]]

smoker.plot(kind='scatter',x='total_bill',y='tip',label="吸烟者")
smoker1.plot(kind='scatter',x='total_bill',y='tip',marker="<",color="red",label="非吸烟者")
plt.show()

往期推荐

django框架学习之分页

机器学习之绘图(一)

爬虫学习之selenium