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

我们的任务:吸烟顾客与不吸烟顾客的消费金额与小费绘制散点图

话不多说 上代码

首先引入需要用到的模块

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