#9、mat基础2

import matplotlib.pyplot as plt
import numpy as np
from mpl_toolkits.mplot3d import Axes3D

x = np.linspace(-3,3,50)  #从-3到3的范围有50个点
y1 = 2*x+1
y2 = x**2
#11、将图画在不同的窗口里
plt.figure('一')
plt.plot(x,y1,color='red',linewidth=1.0,linestyle='--')
plt.figure('二',figsize(长，宽，高))
plt.plot(x,y2)
plt.show()

#12、绘制直角坐标 gca = 'get current axis'
ax = plt.gca()
    #a.绘制边框
ax.spines['right'].set_color('none')
ax.spines['top'].set_color('none')
    #b.绘制xy轴
ax.xaxis.set_ticks_position('bottom')
ax.yaxis.set_ticks_position('left')
    #c.挪动坐标轴
ax.spines['bottom'].set_position(('data',-1))#data处可填outward、axes...
ax.spines['left'].set_position(('data',-1))

#13、做一个图例
l1 = plt.plot(x,y1,color='red',linewidth=1.0,linestyle='--')
l2 = plt.plot(x,y2)
plt.legend(handles=[l1,l2],labels=['线1','线2'],loc='best')

#14、注解
    #a.画点
x0 = 1
y0 = 2*x0 + 1
plt.scatter(x0,y0,s=50,color='b')
    #b.对应虚线
plt.plot([x0,y0],[y0,0],'k--',lw=2.5)#标准码，可套用
    #c.注释（把参数建个字典）
plt.annotate(r'$good$'%y0,xy=(x0,y0),xycoords='data',\
             xytext=(+30.-30),textcoords='offset points',fontsize=16,\
             arrowprops=dict(arrowstyle='->',connectionstyle='arc3,rad=0.2'),ha='center',va='top')
plt.text(第二种方法,同上)
'''注释的文本，位置坐标，坐标是以谁为参考定的，文本显示的位置，xx，\
 字大小，箭头,条形图中horizontal/vertical alignment对齐'''


#15、坐标处理
for label in ax.get_xticklabels() + ax.get_yticklabels():
    label.set_fontsize(12)
    label.set_bbox(dict(facecolor='white',edgecolor='None',alpha=0.7)) #alpha为透明度

#16、等高线图
    #a.高度计算公式
def f(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)
    #b.底图网格
X,Y = np.meshgrid(x,y)
    #c.把颜色填充进去
plt.contourf(X,Y,f(X,Y),8,alpha=0.75,cmap=plt.cm.hot) #cmap把线与颜色对应 8为把等高线分成几个圈可以换
    #d.画线
C = plt.contour(X,Y,f(X,Y),8,color='black',lw=.5) #cmap把线与颜色对应
    #e.加label
plt.clabel(C,inline=True,fontsize=10)  #True 数字在线中间，False 线从数字间穿过

#17、3D图像
#http://matplotlib.org/examples/images_contours_and_fields/interpolation_methods.html
    #a.建立3D系
fig = plt.figure()
ax = Axes3D(fig)
    #b.X Y Z坐标轴
X = np.range(-4,4,0.25)
Y = np.range(-4,4,0.25)
R = np.sqrt(X**2 + Y**2)  #x方加y方开根号
Z = np.sin(R)    #Z轴
    #填色绘图
ax.plot_surface(X, Y, Z, cmap=cm.rainbow,rstride=1,cstride=5
                       linewidth=0, antialiased=False)#r s stride为行列跨度
plt.contourf(X,Y,Z,zdir='z',offset=-2,cmap=plt.get_cmap('rainbow'))#zdir=z为把3D图投影到z轴
ax.set_zlim(-2,2)

#让图显示中文
#coding:utf-8
#有中文出现的情况，需要u'内容'
plt.rcParams['font.sans-serif']=['SimHei'] #用来正常显示中文标签
plt.rcParams['axes.unicode_minus']=False #用来正常显示负号