matplotlib的使用

2018-01-29  本文已影响0人  jshan

matplotlib.pyplot 绘制线性直线

例如

import matplotlib.pyplot as plt
x1 = range(1, 5)
y1 = range(5, 9)
x2 = range(2, 6)
y2 = [0, 3, 5, 6]
plt.plot(x1, y1, label='line one')
plt.plot(x2, y2, label='line two')
plt.legend(loc=0)  # 显示曲线的label说明,loc参数表示label的显示位置,例如0表示best space
plt.xlabel('x_data')  # 图形的x轴
plt.ylabel('y_data')  # 图形的y轴
plt.title('title is here!')  # 图形的标题

plt.show()  # 在同一个画布上显示

柱形图、直方图、散点图的绘制

import matplotlib.pyplot as plt

# 柱状图
plt.bar(x, y)

# 设置刻度范围
plt.axis([x_start, x_end, y_start, y_end])  # e.g plt.axis([0, 10, 0, 12])

# 直方图
x = np.random.randint(1, 100, 100)  # (start, end, total)
bins = range(0, 100, 10)
plt.hist(x, bins)

# 散点图
x = np.random.randint(1, 10, 50)
y = np.random.randint(1, 10, 50)
plt.scatter(x, y)
## 也可以使用多组图形来显示
x1 = np.random.randint(1, 10, 50)
y1 = np.random.randint(1, 10, 50)
plt.scatter(x, y, x1, y1)
plt.show()

面向对象形式的绘图与饼图的绘制

# 面向对象的方式
fig, ax = plt.subplots()  # 初始化绘图对象 fig是画布 ax是图像
ax.scatter(x, y)
plt.show()

# 饼图
fig, ax = plt.subplots()
size = [10, 20, 35, 23]
label = ['a', 'b', 'c', 'd']
explode = (0, 0.1, 0, 0)  # 用于表示图形哪一块突出来的像素,例如这里是表示'b'突出来0.1个像素
ax.pie(size, labels=label, autopct='%1.1f%%', shadow=True, startangle=90, explode=explode)  
# autopct 表示显示百分比;shadow 表示阴影;startangle 表示'a'从那里开始绘图,90度开始,abcd逆时针绘图
ax.axis('equal')  # 表示显示的是一个正圆
plt.show()

绘制多个子图

# 绘制三个子图
x1 = np.linspace(0.0, 5.0)
x2 = np.linspace(0.0, 2.0)
x3 = np.linspace(0.0, 10.0)

y1 = np.cos(2 * np.pi * x1) * np.exp(-x1)
y2 = np.cos(2 * np.pi * x2)
y3 = x3 ** 2 + 2

plt.subpot(2, 2, 1)
plt.plot(x1, y1, 'o-')
plt.title('cos(2*pie*x) * exp(-x)')

plt.subplot(2, 2, 2)
plt.plot(x2, y2, '.-')
plt.tilte('cos(2*pie*x)')

plt.subplot(2, 1, 2)
plt.plot(x3, y3, '^-')
plt.title('x^2 + 2')

plt.show()
# 
mu, sigma = 100, 15
x = mu + sigma * np.random.randn(10000)  # 表示生成一组满足正太分布N(mu, sigma^2)的10000个浮点数据
n, bins, patches = plt.hist(x, 50, normed=1, facecolor='g', alpha=0.75)

绘制3D图像

#
fig = plt.figure()
ax1 = fig.add_subplot(111, projection='3d')
x = []
y = []
z = []

ax1.scatter()


# 二次抛物面 z=x^2+y^2
x = np.linspace(-10, 10, 10)
y = x
x, y = np.meshgrid(x, y)  # 生成数据
z = x ** 2 + y ** 2
ax = plt.subplot(111, projection='3d')
ax.plot_wireframe(x, y, z)  # 画线图
plt.show()

使用中文

plt.rcParams['font.sans-serif'] = ['SimHei']  # 用来正常显示中文标签
plt.rcParams['axes.unicode_minus'] = False  # 用来正常显示负号

# 例子
x = [1, 2, 3, 4]
y = [5, 6, 7, 1]

x1 = [1, 2, 3, 4]
y1 = [9, 4, 5, 1]
plt.plot(x, y, label='x')
plt.plot(x1, y1, label='y')
plt.xlabel('x坐标')
plt.ylabel('y坐标')
plt.title('例子')
plt.legend(loc=0)  # 可选位置参数loc,0是最优

# 叠加图
t = np.arange(0., 5., 0.2)
## red dashes, blue squares and green triangles
### 可以在一个图标中画多条图形,另外可以用简写来画图,例如 'r--'
plt.plot(t, t, 'r--', t, t**2, 'bs', t, t**3, 'g^')
plt.show()

# 练习
x = np.linspace(-np.pi, np.pi, 256)
plt.plot(x, np.sin(x), color='blue', linestyle='-', lw=2, label='sin line')
plt.plot(x, np.cos(x), 'r--', label='cos line')
plt.show()

## 调整坐标范围
plt.xlim(-4, 4)
plt.ylim(-1.1, 1.1)

## 调整坐标刻度 第一个列表是位置,第二个列表是位置对应的值
plt.xticks([-np.pi, -np.pi/2, 0, np.pi/2, np.pi], [r'$-\pi$', r'$-\pi/2$', r'$0$', r'$\pi/2$', r'$\pi$'])
plt.yticks([-1, 0, 1])

## 调整坐标中横纵轴的位置

ax = plt.gca()  # 获取当前图标, get current axis
ax.spines['right'].set_color('none')  #先把右边和上边的边界设置为不可见
ax.spines['top'].set_color('none')
ax.xaxis.set_ticks_position('bottom')
ax.yaxis.set_ticks_position('left')
ax.spines['bottom'].set_position(('data', 0))  # 把下边界和左边界移动到0点
ax.spines['left'].set_position(('data', 0))
plt.legnd()

plt.show()

绘制colormap图形

# 这里绘制 sklearn.datasets.load_digits() 中手写数字的图形
from sklearn import datasets
import matplotlib.pyplot as plt

digits = datasets.load_digtis()
plt.figure(1, figsize=(3, 3))
# 这里绘制最后一个图形,并且采用彩色图形,颜色是灰色gray,gray_r表示,插值采用临近方式
plt.imshow(digits.images[-1], cmap=plt.cm.gray_r, interpolation='nearest')  
plt.show()

# 这里说明一下 interpolation 插值的使用
- None 那么就是使用配置 rc 中的参数 image.interpolation 
- "none" 那么就没有任何interpolation设置在这个图片上
- 其他的默认都是 "neareast"

# 关于 cmap 的颜色设置可以参考如下链接 http://www.cnblogs.com/denny402/p/5122594.html
上一篇 下一篇

猜你喜欢

热点阅读