import numpy as np
import pandas as pd

x = np.arange(-2.0,2.0,0.01)
y = np.arange(-2.0,2.0,0.01)
'''meshgrid用于生成三维曲面的分格线座标；产生“格点”矩阵'''
X,Y = np.meshgrid(x,y)  # 确定x/y的取值范围
'''定义一个函数，用来计算x/y对应的z值'''
def f(x,y):
    return (1-y**5+x**5)*np.exp(-x**2-y**2)
'''contour()函数可生成三维结构表面的等值线图'''
C = plt.contour(X,Y,f(X,Y),8,colors='black')
'''clabel用于标记等高线'''
plt.clabel(C,inline=1,fontsize=10)
plt.show()

x = np.arange(-2.0,2.0,0.01)
y = np.arange(-2.0,2.0,0.01)
'''meshgrid用于生成三维曲面的分格线座标；产生“格点”矩阵'''
X,Y = np.meshgrid(x,y)  # 确定x/y的取值范围
'''定义一个函数，用来计算x/y对应的z值'''
def f(x,y):
    return (1-y**5+x**5)*np.exp(-x**2-y**2)
'''contour()函数可生成三维结构表面的等值线图'''
C = plt.contour(X,Y,f(X,Y),8,colors='black')

'''cmap=plt.cm.hot为等值线添加过渡色'''
plt.contour(X,Y,f(X,Y),8,cmap=plt.cm.hot)

'''clabel用于标记等高线'''
plt.clabel(C,inline=1,fontsize=10)
plt.show()

x = np.arange(-2.0,2.0,0.01)
y = np.arange(-2.0,2.0,0.01)
'''meshgrid用于生成三维曲面的分格线座标；产生“格点”矩阵'''
X,Y = np.meshgrid(x,y)  # 确定x/y的取值范围
'''定义一个函数，用来计算x/y对应的z值'''
def f(x,y):
    return (1-y**5+x**5)*np.exp(-x**2-y**2)
'''contour()函数可生成三维结构表面的等值线图'''
C = plt.contour(X,Y,f(X,Y),8,colors='black')
'''cmap=plt.cm.hot为等值线添加过渡色'''
plt.contour(X,Y,f(X,Y),8,cmap=plt.cm.hot)
'''clabel用于标记等高线'''
plt.clabel(C,inline=1,fontsize=10)

'''colorbar()可在右侧显示颜色值'''
plt.colorbar()

plt.show()

N = 8
'''生成角度值'''
theta = np.arange(0.,2*np.pi,2*np.pi/N)
'''生成半径值'''
radii = np.array([4,7,5,3,1,5,6,7])
'''定义轴类型'''
plt.axes([0.025,0.025,0.95,0.95],polar=True)
'''定义颜色集，这里使用RGB值，当然也可以使用颜色名称'''
colors = np.array(['#4bb2c5','#c5b47f','#EAA228','#579575','#839557','#958c12','#953579','#4b5de4'])
'''bar()函数要求传入角度和半径作为参数'''
bars = plt.bar(theta,radii,width=(2*np.pi/N),bottom=0.0,color=colors)
plt.show()