散点图的曲线绘制：散点图的绘制

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上的开源地址：https://github.com/jackzhenguo/python-small-examples

欢迎关注微信公众号【Python与算法社区】