plt分类着色

Create as many colors as there are unique midwest[‘category’]

categories = np.unique(midwest[‘category’]) # 取 category 列的所有元素的集合
colors = [plt.cm.tab10(i/float(len(categories)-1)) for i in range(len(categories))]

Draw Plot for Each Category

plt.figure(figsize=(16, 12), dpi= 50, facecolor=‘w’, edgecolor=‘k’)

for i, category in enumerate(categories):
plt.scatter(‘area’, ‘poptotal’,
data=midwest.loc[midwest.category==category, :],
s=50, cmap=colors[i], label=str(category))# s 控制点的大小