python 散点图_Python绘制散点图

import numpy as np

import pandas as pd

import matplotlib as mpl

import matplotlib.pyplot as plt

# 版本检查

print(mpl.__version__) #> 3.0.0

# 导入数据集

midwest = pd.read_csv("https://raw.githubusercontent.com/selva86/datasets/master/midwest_filter.csv")

# midwest['category']分类，颜色设置为与其一样多

categories = np.unique(midwest['category'])

colors = [plt.cm.tab10(i/float(len(categories)-1)) for i in range(len(categories))]

# 每个分类plot

plt.figure(figsize=(16, 10), dpi= 80, facecolor='w', edgecolor='k')

for i, category in enumerate(categories):

plt.scatter('area', 'poptotal',

data=midwest.loc[midwest.category==category, :],

s=20, c=colors[i], label=str(category))

# 修改x轴,y轴坐标系尺寸区间

plt.gca().set(xlim=(0.0, 0.1), ylim=(0, 90000),

xlabel='Area', ylabel='Population')

plt.xticks(fontsize=12)

plt.yticks(fontsize=12)

plt.title("Midwest Area vs Population", fontsize=22)

plt.legend(fontsize=12)

plt.show()

本案例在Github上的开源链接为：

jackzhenguo/python-small-examples​github.com

欢迎关注公众号【Python与算法社区】，点击下方链接，一键获取273篇原创学习笔记

认真就能打动人：273篇干货资料汇总​mp.weixin.qq.com