python中matplotlib 的简单使用

 

1、简单折线图的画图，轴标签、图的颜色，风格，等等参数，本文只介绍最常用的几个参数：

import matplotlib.pyplot as plt
import numpy as np

x = np.linspace(-3,3,50)
y1 = 2 * x + 1
y2 = x ** 2

plt.figure()  # figure 下面的图形都出现再这个figure中
plt.plot(x,y2,label = 'up')   #label 为该图像的图例
plt.plot(x,y1,color = 'red',linewidth = 1.0,linestyle = '--',label = 'down')
plt.xlim((-1,2))  # 对x轴的范围限制
plt.ylim((-2,3))
plt.xlabel('I am X')  # x轴的标签
plt.ylabel('I am Y')

# 边框删除
ax = plt.gca()  # 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)) # 左边移动到 0


# 图例
plt.legend()  # 打印出画图时加入的 label 参数

#注释
x0 = 1
y0 = 2 * x0 +1
plt.scatter(x0,y0,s = 50,color = 'b')   # 找到要进行注释的一个坐标点
# plt.plot([x0,x0],[y0,0],'k--',linewidth = 2.5)  # 画一条竖直线

# plt.annotate(r'$y=2x+1$',xy=(x0,y0),xycoords='data',xytext=(+30,-30),textcoords='offset points')

plt.text(-1,2,r'$This\ is\ \mu\ \sigma_i\ \alpha_t$',
         fontdict={'size':16,'color':'r'}) # 数学符号前面加\转义符，显示数学符号

plt.show()

 2、散点图

import matplotlib.pyplot as plt
import numpy as np

n = 1024
X = np.random.normal(0,1,n)
Y = np.random.normal(0,1,n)
# 散点图
T = np.arctan2(Y,X) # 颜色

plt.scatter(X,Y,s=75,c=T,alpha=0.5) # alpha 表示透明度
plt.xlim((-1.5,1.5))
plt.ylim((-1.5,1.5))

plt.show()

3、柱状图

# 柱状图
n = 12
X = np.arange(n)
Y1 = (1 - X / float(n)) * np.random.uniform(0.5,1.0,n)
Y2 = (1 - X / float(n)) * np.random.uniform(0.5,1.0,n)

plt.bar(X,+Y1,facecolor='b',edgecolor = 'red')
plt.bar(X,-Y2)

for x,y in zip(X,Y1): # 在柱状图上加上高度值
    plt.text(x,y+0.05,'%.2f'%y,ha = 'center',va='bottom')#横向，纵向对齐方式

plt.show()

 4、等高线

# 等高图
import matplotlib.pyplot as plt
import numpy as np

def f(x,y): # 计算点（x,y)处的高度
    return (1 - x / 2 + x ** 5 + y ** 3) * np.exp(-x ** 2 - y ** 2)

n = 256
x = np.linspace(-3,3,n)
y = np.linspace(-3,3,n)

X,Y = np.meshgrid(x,y)

# use plt.contourf to filling contours
# X,Y and value for (X,Y) point

plt.contourf(X,Y,f(X,Y),8,alpha=0.75,cmap=plt.cm.hot)# 8 是分成几个高度

# use plt.contour to add contour lines  contour:等高线
C = plt.contour(X,Y,f(X,Y),8,color='black',linewidth=0.5)

# adding label
plt.clabel(C,inline=True,fontsize=10) # 标签在线里面

plt.show()