import numpy as np import matplotlib.pyplot as plt def main(): #scatter fig = plt.figure() ax = fig.add_subplot(3,3,1) n = 128 X = np.random.normal(0,1,n) Y = np.random.normal(0,1,n) T = np.arctan2(Y,X) #plt.axes([0.025,0.025,0.95,0.95]) ax.scatter(X,Y,s=75,c=T,alpha=.5) plt.xlim(-1.5,1.5), plt.xticks([]) plt.ylim(-1.5,1.5), plt.yticks([]) plt.axis() plt.title('scatter') plt.xlabel('x') plt.ylabel('y') #bar fig.add_subplot(332) n =10 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='#9999ff', edgecolor='white') plt.bar(X, -Y2, facecolor='#ff9999', edgecolor='white') for x,y in zip(X,Y1): plt.text(x + 0.4, y + 0.05,'%.2f' %y,ha='center', va = 'bottom') for x,y in zip(X,Y2): plt.text(x+0.4,-y - 0.05, '%.2f' %y, ha='center',va ='top') #Pie fig.add_subplot(333) n = 20 Z = np.ones(n) Z[-1] *=2 plt.pie(Z,explode=Z * .05,colors=['%f' % (i / float(n)) for i in range(n)], labels=['%.2f' % (i/float(n)) for i in range(n)]) plt.gca().set_aspect('equal') plt.xticks([]),plt.yticks([]) #polar fig.add_subplot(334,polar =True) n = 20 theta = np.arange(0.0, 2* np.pi, 2*np.pi /n) radii = 10 * np.random.rand(n) # plt.plot(theta, radii) plt.polar(theta,radii) #heatmap fig.add_subplot(335) from matplotlib import cm data = np.random.rand(3,3) # print(data) cmap =cm.Blues map= plt.imshow(data,interpolation='nearest',cmap=cmap,aspect='auto',vmin=0,vmax=1) #3D from mpl_toolkits.mplot3d import Axes3D ax = fig.add_subplot(336,projection='3d') ax.scatter(1,1,3,s =100) #hot map fig.add_subplot(313) 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) X,Y = np.meshgrid(x,y) plt.contourf(X,Y,f(X,Y),8,alpha=.75,cmap=plt.cm.hot) plt.savefig('D:/fig.png') plt.show() if __name__ == '__main__': main()