Python绘制高清论文图件

原代码

以下面的代码生成一个模型，并画图显示

nx, nz = 201, 101
data = np.zeros((nx, nz))
data[:, 40] = 20
data[:, 40:70] = 30
data[:, 70:] = 40
plt.imshow(data.T)
plt.show()

结果是这样的

显然是不能放进论文里的，除了缺失x、y轴标签，还有重要的一点，其坐标轴刻度表示的是所绘制数据的尺寸，假设模型dx=dz=5m，则刻度上的数字都应该乘5。并且这样生成的图片有很大的白边，需要手动裁切。

改进后的代码

from mpl_toolkits.axes_grid1 import make_axes_locatable
from pyseismic.image_processing import savefig
import matplotlib.pyplot as plt
import numpy as np

plt.rcParams['font.size'] = 10
plt.rcParams['font.family'] = 'Microsoft YaHei'

nx, nz = 201, 101
data = np.zeros((nx, nz))
data[:, 40] = 20
data[:, 40:70] = 30
data[:, 70:] = 40
dx = dz = 5

xtick = list(range(0, nx, 25))
xticklabel = [str(i*dx) for i in xtick]
ytick = list(range(0, nz, 20))
yticklabel = [str(i*dz) for i in ytick]

img_w = 6.36
img_h = 5

fig, ax = plt.subplots(figsize=(img_w, img_h))
im = ax.imshow(data.T)
ax.xaxis.set_ticks_position('top')
ax.xaxis.set_label_position('top')
ax.xaxis.set_ticks(xtick)
ax.xaxis.set_ticklabels(xticklabel)
ax.yaxis.set_ticks(ytick)
ax.yaxis.set_ticklabels(yticklabel)
ax.set_xlabel('水平距离（m）', labelpad=5)
ax.set_ylabel('深度（m）')
divider = make_axes_locatable(ax)
cax = divider.append_axes('bottom', size='5%', pad=0.05)
plt.colorbar(im, cax=cax, orientation='horizontal')

savefig(fig, 'data.tif', dpi=300)
plt.show()

插入word中并居中，效果如下：

 注意

【figsize】在word中查看布局-页面设置，可计算出正文中图片最大宽度可设置为5.76英寸，但因为matplotlib画图的figsize包含了需要裁切掉的白边，故代码中设置figsize宽度为6.36，这个需要根据实际情况调整。

【savefig】改进的代码使用了我编写的pyseismic库中的savefig函数，可以自动裁切图片白边，设置dpi=300可保存高清图片，图片后缀设为.tif可有效减少图片大小。