python可视化之散点图绘制

---

一、绘制一种颜色的散点图

1.引入库

代码如下

import numpy as np
import matplotlib.pyplot as plt

2、步骤

创建一组数据

x1 = np.random.randn(10)
x2 = x1 + x1**2 - 10

确定一个画布，当只有一个图的时候，这个不是必须的

plt.figure(figsize=(8,4))

绘图，s是散点的size，c为color

plt.scatter(x1,x2,s=100,c="yellow",label = "Positive")

显示图形

plt.legend()
plt.show()

二、绘制多个标签多种颜色的散点图

1.引入库

代码如下

import numpy as np
import matplotlib.pyplot as plt

2、步骤

创建一组数据，并分类

x = np.random.randn(10,2)
y = np.array([0,0,1,1,0,1,0,1,0,0])

确定一个画布，当只有一个图的时候，这个不是必须的

plt.figure(figsize=(8,4))

绘图

colors = ["red","black"]
labels = ["zero","one"]
for i in range(x.shape[1]):
    plt.scatter(x[y==i,0],x[y==i,1],c=colors[i],label = labels[i])

显示图形

plt.legend()
plt.show()

二、绘制大量标签多种颜色的散点图

代码如下

midwest = pd.read_excel(“简易数据.xls”)
plt.figure(figsize=(16,10))
categories = np.unique(midwest[‘简称’])

for i in range(len(categories)):
plt.scatter(midwest.loc[midwest[“简称”] == categories[i],‘24H涨跌幅’]
,midwest.loc[midwest[“简称”] == categories[i],‘24H成交额’]
,s = 20
,c= np.array(plt.cm.tab10(i/len(categories))).reshape(1,-1)
,label = categories[i])
plt.legend()
plt.show()