import matplotlib.pyplot as plt
import numpy as np
x = np.linspace(-1,1,100) #從-1 到1生成100個帶你
y = 2*x +1
plt.plot(x,y)
plt.show()
figure 圖像
x = np.linspace(-1,1,100)
y1 = 2x +2
y2 = x*2
plt.figure()
plt.plot(x,y1)
plt.figure(figsize=(8,2)) #設置圖像的大小
plt.plot(x,y2)
plt.show()
將兩個圖畫在一起
plt.plot(x,y1,color='red',linewidth=1.0,linestyle='--')
plt.plot(x,y2,color='blue',linewidth=5.0,linestyle='-')
plt.show()
設計坐標軸
x = np.linspace(-1,1,100)
y1 = 2x +2
y2 = x*2
plt.xlim((-1,2))
plt.ylim((-2,3)) #限制xy的范圍
plt.xlabel('I am X')
plt.ylabel('I am Y') #設置xy的名稱
plt.plot(x,y1,color='red',linewidth=1.0,linestyle='--')
plt.plot(x,y2,color='blue',linewidth=5.0,linestyle='-')
plt.show()
使用自定義的xy的坐標軸1
x = np.linspace(-3,3,100)
y1 = 2x +2
y2 = x*2
plt.xlim((-1,2))
plt.ylim((-2,3)) #限制xy的范圍
plt.xlabel('I am X')
plt.ylabel('I am Y') #設置xy的名稱
plt.plot(x,y1,color='red',linewidth=1.0,linestyle='--')
plt.plot(x,y2,color='blue',linewidth=5.0,linestyle='-')
new_ticks = np.linspace(-2,2,11)
print(new_ticks)
plt.xticks(new_ticks)
plt.yticks([-1,0,1,2,3],
['leverl1','leverl2','leverl3','leverl4','leverl5'])
plt.show()
自定義xy的坐標軸2
x = np.linspace(-3,3,100)
y1 = 2x +1
y2 = x*2
plt.xlim((-1,2))
plt.ylim((-2,3)) #限制xy的范圍
plt.xlabel('I am X')
plt.ylabel('I am Y') #設置xy的名稱
plt.plot(x,y1,color='red',linewidth=1.0,linestyle='--')
plt.plot(x,y2,color='blue',linewidth=5.0,linestyle='-')
new_ticks = np.linspace(-2,2,11)
print(new_ticks)
plt.xticks(new_ticks)
plt.yticks([-1,0,1,2,3],
['leverl1','leverl2','leverl3','leverl4','leverl5'])
gca get current axis 獲取當前的軸
ax = plt.gca()
ax.spines['right'].set_color('none') #自定義右邊的框 若要設置顏色將none改為red 或其他
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)) #將x軸的位置放到數據0的位置
ax.spines['left'].set_position(('data',0)) #將y軸的位置放到數據0的位置
plt.show()
圖例的設置
x = np.linspace(-3,3,100)
y1 = 2x +1
y2 = x*2
plt.xlim((-1,2))
plt.ylim((-2,3)) #限制xy的范圍
plt.xlabel('I am X')
plt.ylabel('I am Y') #設置xy的名稱
l1,=plt.plot(x,y1,color='red',linewidth=1.0,linestyle='--')
l2,=plt.plot(x,y2,color='blue',linewidth=5.0,linestyle='-')
plt.legend(handles=[l1,l2],labels=['test1','test2'],loc='best')
new_ticks = np.linspace(-2,2,11)
print(new_ticks)
plt.xticks(new_ticks)
plt.yticks([-1,0,1,2,3],
['leverl1','leverl2','leverl3','leverl4','leverl5'])
plt.show()
給圖片注釋
x = np.linspace(-1,1,100)
y1 = 2x +1
y2 = x*2
plt.xlim((-1,2))
plt.ylim((-2,3)) #限制xy的范圍
plt.xlabel('I am X')
plt.ylabel('I am Y') #設置xy的名稱
plt.plot(x,y1,color='red',linewidth=1.0,linestyle='-')
new_ticks = np.linspace(-2,2,11)
print(new_ticks)
plt.xticks(new_ticks)
plt.yticks([-1,0,1,2,3],
['leverl1','leverl2','leverl3','leverl4','leverl5'])
gca get current axis 獲取當前的軸
ax = plt.gca()
ax.spines['right'].set_color('none') #自定義右邊的框 若要設置顏色將none改為red 或其他
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)) #將x軸的位置放到數據0的位置
ax.spines['left'].set_position(('data',0)) #將y軸的位置放到數據0的位置
x0 = 0.5
y0 = 2*x0 + 1
plt.scatter(x0,y0,s=50,color='b') #畫點
plt.plot([x0,x0],[y0,0],'k--',lw=2) #畫虛線 k--表示黑虛線 lw線寬
plt.annotate(r'' % y0,xy=(x0,y0),xytext=(+30,-30),textcoords='offset points',fontsize=16,
arrowprops=dict(arrowstyle='->',connectionstyle='arc3,rad=.2'))
plt.text(-1,2.5,r'',fontdict={'size':'16','color':'r'})
plt.show()
散點圖
x = np.random.normal(0,1,500)
y = np.random.normal(0,1,500)
plt.scatter(x,y,s=50,c='b',alpha=0.5)
plt.xlim((-2,-2))
plt.ylim((-2,-2))
plt.xticks(())
plt.yticks(()) #去掉軸的標注
plt.show()
直方圖
x1= np.arange(10)
y1= 2**x1 + 10
plt.bar(x1,y1,facecolor='#9999ff',edgecolor='white') #為-y則是往下畫直方圖
for x1,y1 in zip(x1,y1): #zip是同時讀取x y
plt.text(x1,y1,'%.2f' % y1,ha='center',va='bottom')
plt.show()
繪制等高線圖
def f(x,y):
"""
計算高度的函數
:param x: 向量
:param y: 向量
:return: dim(x)dim(y)維的矩陣
"""
# the height function
return (1 - x / 2 + x5 + y3) * np.exp(-x2 -y*2)
x = np.linspace(-3,3,100)
y = np.linspace(-3,3,100)
X,Y = np.meshgrid(x,y) # 獲得網格坐標矩陣
plt.contourf(X,Y,f(X,Y),8,cmap=plt.cm.hot) # 進行顏色填充
c = plt.contour(X,Y,f(X,Y),8,colors='black') # 進行等高線繪制
plt.clabel(c,inline=True,fontsize=10) # 線條標注的繪制
plt.xticks(())
plt.yticks(())
plt.show()
繪制3d圖 將代碼復制到 ipython中shift ＋回車鍵 查看3d圖
from mpl_toolkits.mplot3d import Axes3D
fig = plt.figure()
ax = Axes3D(fig)
x = np.arange(-4,4,0.2)
y = np.arange(-4,4,0.2)
X,Y = np.meshgrid(x,y)
R = np.sqrt(X2+Y2)
Z = np.sin(R)
cmap = plt.get_cmap('rainbow')
ax.plot_surface(X,Y,Z,rstride=1,cstride=1,cmap=cmap)
ax.contourf(X,Y,Z,zdir='z',offset=-2.0020,cmap='rainbow')
ax.set_zlim(-2,-2)
plt.show()
繪制subplot
plt.figure()
plt.subplot(2,2,1) #兩行兩列的圖 這個放第一個位置 或者去掉括號 （221）
plt.plot([0,1],[0,1])
plt.subplot(2,2,2)
plt.plot([0,1],[0,1])
plt.subplot(2,2,3)
plt.plot([0,1],[0,1])
plt.subplot(2,2,4)
plt.plot([0,1],[0,1])
plt.show()
大小不一的圖片
plt.figure()
plt.subplot(2,1,1) #兩行兩列的圖 這個放第一個位置 或者去掉括號 （221）
plt.plot([0,1],[0,1])
plt.subplot(2,3,4)
plt.plot([0,1],[0,1])
plt.subplot(2,3,5)
plt.plot([0,1],[0,1])
plt.subplot(2,3,6)
plt.plot([0,1],[0,1])
plt.show()
動態圖
from matplotlib import animation
import matplotlib.pyplot as plt
import numpy as np
fig,ax = plt.subplots()
x= np.arange(0,2*np.pi,0.01)
line,= ax.plot(x,np.sin(x))
def animate(i):
line.set_ydata(np.sin(x+i/10))
return line,
def init():
line.set_ydata(np.sin(x))
return line,
ani = animation.FuncAnimation(fig=fig,func=animate,init_func=init,interval=20)
plt.show()
