Matplotlib笔记二 图的样式

只需要修改x,y 方便画

2.1散点图

import matplotlib.pyplot as plt
import numpy as np

n = 1024    
X = np.random.normal(0, 1, n)#正态分布 平均0 方差1 1024个
Y = np.random.normal(0, 1, n)
T = np.arctan2(Y, X)#单纯为了颜色好看

plt.scatter(X, Y, s=75, c=T, alpha=.5) #x,ysize,color,透明度

plt.xlim(-1.5, 1.5)
plt.xticks(())  # 去掉x轴刻度
plt.ylim(-1.5, 1.5)
plt.yticks(())  # 去掉y轴刻度

plt.show()

2.2柱状图

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='#9999ff', edgecolor='white')
plt.bar(X, -Y2, facecolor='#ff9999', edgecolor='white')

# 对于每一个点（x,y）加 text，位置，保留两位小数y，位置
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')

plt.show()

2.3等高线图

n = 256
x = np.linspace(-3, 3, n)
y = np.linspace(-3, 3, n)
X,Y = np.meshgrid(x, y)# 把x,y生成网格

plt.contourf(X, Y, f(X, Y), 8, alpha=.75, cmap=plt.cm.hot)#设置颜色  前三个参数X，Y，Z
#数字8 把等高线分成10部分
C = plt.contour(X, Y, f(X, Y), 8, colors='black', linewidth=.5)#画等高线
plt.clabel(C, inline=True, fontsize=10)#添加label 数字描述

plt.show()

2.4image图像

把数字用颜色显示出
#之前做关系矩阵和roi模拟用到过

a = np.array([0.313660827978, 0.365348418405, 0.423733120134,
              0.365348418405, 0.439599930621, 0.525083754405,
              0.423733120134, 0.525083754405, 0.651536351379]).reshape(3,3)
plt.imshow(a, interpolation='nearest', cmap='bone', origin='lower')
#origin的lower和upper是上下颠倒的
plt.colorbar(shrink=.92)     
plt.show()     

参数interpolation说明：http://matplotlib.org/examples/images_contours_and_fields/interpolation_methods.html

2.5 3d图

# 要加上
from mpl_toolkits.mplot3d import Axes3D

fig = plt.figure()
ax = Axes3D(fig)#加上3d坐标轴
X = np.arange(-4, 4, 0.25)
Y = np.arange(-4, 4, 0.25)
X, Y = np.meshgrid(X, Y) #固定这个形式
R = np.sqrt(X ** 2 + Y ** 2)
# height value
Z = np.sin(R) #Z值
ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap=plt.get_cmap('rainbow')，edgecolor='black')# r和c参数 代表跨度 数值越大 跨度越大 网格越大
ax.contourf(X, Y, Z, zdir='z', offset=-2, cmap=plt.get_cmap('rainbow'))#画等高线图 从z轴压下去 offset代表压倒 z=-2的位置
ax.set_zlim(-2, 2)
plt.show()