基于散点图(scatter)的基础上稍加调整就可以变成气泡图。
#载入两个库
import pandas as pd
import matplotlib.pyplot as plt
在Excel上写点数据,这里给个示例
绘制散点图:以数量列为横坐标x,以大小列为纵坐标y,此基础上以大小列为气泡大小
#先定义气泡大小,rank 函数将大小列进行大小分配,越大的值分配结果也越高
#n 为倍数,用来调节气泡的大小,且看后头
size=data['大小'].rank()
n=20
#开始作图
plt.scatter(data['数量'],data['大小'],s=size*n,alpha=0.6)
plt.show()
结果:
补充:数据中还有一个分类,很多时候,我们需要根据分类来对数据点进行区分,这个时候就需要对颜色进行定义
(注:分类列中的数值有:0,1,2项)
#定义一个字典,将颜色跟对应的分类进行绑定
color={0:'red',1:'blue',2:'orange'}
#增加color的参数,用列表解析式将data分类中的每个数据的数字映射到前面color的颜色中
plt.scatter(data['数量'],data['大小'],color=[color[i] for i in data['分类']],s=size*n,alpha=0.6)
plt.show()
结果:
在柱形图中,如果需要对最大或最小项,或其他的项进行不同颜色标注,同样可以用列表解析式来完成。