【python】python绘制两组数据的散点图并添加图例

导入需要用到的包

import numpy as np
import matplotlib.pyplot as plt
import matplotlib as mat

避免中文乱码

mat.rcParams['font.sans-serif'] = ['SimHei']
mat.rcParams['axes.unicode_minus'] = False
a=2

随机生成数据a

x_a=np.random.randn(a)
y_a=np.random.randn(a)

随机生成数据b

b=20
x_b=np.random.randn(b)
y_b=np.random.randn(b)

绘制数据a的散点图

plt.scatter(x_a,y_a,c='r',marker='o')

绘制数据b的散点图

plt.scatter(x_b,y_b,c='g',marker='>')

添加图例

plt.legend(['data a','data b'])

显示散点图

plt.show()

散点图结果

完整代码如下

import numpy as np
import matplotlib.pyplot as plt
import matplotlib as mat
mat.rcParams['font.sans-serif'] = ['SimHei']
mat.rcParams['axes.unicode_minus'] = False
a=2
x_a=np.random.randn(a)
y_a=np.random.randn(a)
b=20
x_b=np.random.randn(b)
y_b=np.random.randn(b)
plt.scatter(x_a,y_a,c='r',marker='o')
plt.scatter(x_b,y_b,c='g',marker='>')
plt.legend(['data a','data b'])
plt.show()