python matplotlib二维地震数据画图

最新推荐文章于 2024-11-13 17:24:18 发布

某崔同学

最新推荐文章于 2024-11-13 17:24:18 发布

阅读量956

点赞数 4

文章标签： python matplotlib numpy

本文链接：https://blog.csdn.net/weixin_49322652/article/details/132765515

版权

本文展示了使用Matplotlib库进行图像数据的预处理，包括原始数据、噪声数据、去噪后的数据以及局部相似性地图的可视化。通过seismic和jetcmap展示处理过程。

摘要由CSDN通过智能技术生成

效果图如上

附代码：

    
def show(x,y,x_,xx):
    import matplotlib.pyplot as plt
    fig = plt.figure(figsize=(10,8))
    #plt.colorbar(shrink= 0.8)
    plt.subplot(221)
    plt.imshow(x, cmap=plt.cm.seismic, vmin=-1, vmax=1)
    # plt.imshow(x, cmap=plt.cm.gray, vmin=-0.5, vmax=0.5)
    plt.title('Raw Data', fontsize=18)
    plt.xlabel('Channel',fontsize=16)
    plt.ylabel('Samples',fontsize=16)
    plt.xticks(fontsize=16)
    plt.yticks(fontsize=16)
    plt.annotate('(a)', xy=(0, 0), xytext=(-60, -60), fontsize=16, fontweight='bold', va='top')
    plt.colorbar(shrink= 0.5)

    plt.subplot(222)   
    plt.imshow(y, cmap=plt.cm.seismic, vmin=-1, vmax=1)
    plt.title('Noisy Data',fontsize=18)
    plt.xlabel('Channel',fontsize=16)
    plt.ylabel('Samples',fontsize=16)
    plt.xticks(fontsize=16)
    plt.yticks(fontsize=16)
    plt.annotate('(b)', xy=(0, 0), xytext=(-60, -60), fontsize=16, fontweight='bold', va='top')
    #plt.text(np.pi/2, -10, 'SNR=xx dB', fontsize=14,  horizontalalignment='center')
    plt.colorbar(shrink= 0.5)

    plt.subplot(223)   
    plt.imshow(x_, cmap=plt.cm.seismic,  vmin=-1, vmax=1)
    plt.title('Denoised Data',fontsize=18)
    plt.xlabel('Channel',fontsize=16)
    plt.ylabel('Samples',fontsize=16)
    plt.xticks(fontsize=16)
    plt.yticks(fontsize=16)
    plt.annotate('(c)', xy=(0, 0), xytext=(-60, -60), fontsize=16, fontweight='bold', va='top')
    plt.colorbar(shrink= 0.5)
    
    plt.subplot(224)   
    #plt.imshow(xx, cmap=plt.cm.seismic,  vmin=-1, vmax=1)
    plt.imshow(xx,cmap='jet',clim=(0,1))
    plt.title('Local Similarity Map',fontsize=18)
    plt.xlabel('Channel',fontsize=16)
    plt.ylabel('Samples',fontsize=16)
    plt.xticks(fontsize=16)
    plt.yticks(fontsize=16)
    plt.annotate('(d)', xy=(0, 0), xytext=(-60, -60), fontsize=16, fontweight='bold', va='top')
    #plt.text(np.pi/2, -10, 'SNR=xx dB', fontsize=14, horizontalalignment='center')
    #plt.annotate('(d)', xy=(-20, -20), xytext=(200, 200), fontsize=14, fontweight='bold', va='top')
    plt.colorbar(shrink= 0.5)
       
    plt.tight_layout() 
    plt.savefig('data2.png', dpi = 600) 
    plt.show()

def show(x,y,x_,xx):
    import matplotlib.pyplot as plt
    fig = plt.figure(figsize=(8,8))
    #plt.colorbar(shrink= 0.8)
    plt.subplot(221)
    plt.imshow(x, cmap=plt.cm.seismic, vmin=-2, vmax=2)
    # plt.imshow(x, cmap=plt.cm.gray, vmin=-0.5, vmax=0.5)
    plt.title('Clean Data', fontsize=16)
    plt.xlabel('Trace',fontsize=12)
    plt.ylabel('Samples',fontsize=12)
    plt.xticks(fontsize=12)
    plt.yticks(fontsize=12)
    plt.annotate('(a)', xy=(0, 0), xytext=(-80, -60), fontsize=16, fontweight='bold', va='top')
    plt.colorbar(shrink= 0.5)

    plt.subplot(222)   
    plt.imshow(y, cmap=plt.cm.seismic, vmin=-2, vmax=2)
    plt.title('Noisy Data',fontsize=16)
    plt.xlabel('Trace',fontsize=12)
    plt.ylabel('Samples',fontsize=12)
    plt.xticks(fontsize=12)
    plt.yticks(fontsize=12)
    plt.annotate('(b)', xy=(0, 0), xytext=(-80, -60), fontsize=16, fontweight='bold', va='top')
    #plt.text(np.pi/2, -10, 'SNR=xx dB', fontsize=14,  horizontalalignment='center')
    plt.colorbar(shrink= 0.5)

    plt.subplot(223)   
    plt.imshow(x_, cmap=plt.cm.seismic,  vmin=-2, vmax=2)
    plt.title('Denoised Data',fontsize=16)
    plt.xlabel('Trace',fontsize=12)
    plt.ylabel('Samples',fontsize=12)
    plt.xticks(fontsize=12)
    plt.yticks(fontsize=12)
    plt.annotate('(c)', xy=(0, 0), xytext=(-80, -60), fontsize=16, fontweight='bold', va='top')
    plt.colorbar(shrink= 0.5)
    
    plt.subplot(224)   
    plt.imshow(xx, cmap=plt.cm.seismic,  vmin=-2, vmax=2)
    #plt.imshow(xx,cmap='jet',clim=(0,1))
    plt.title('Removed Noise',fontsize=16)
    plt.xlabel('Trace',fontsize=12)
    plt.ylabel('Samples',fontsize=12)
    plt.xticks(fontsize=12)
    plt.yticks(fontsize=12)
    plt.annotate('(d)', xy=(0, 0), xytext=(-90, -60), fontsize=16, fontweight='bold', va='top')
    #plt.text(np.pi/2, -10, 'SNR=xx dB', fontsize=14, horizontalalignment='center')
    #plt.annotate('(d)', xy=(-20, -20), xytext=(200, 200), fontsize=14, fontweight='bold', va='top')
    plt.colorbar(shrink= 0.5)
       
    plt.tight_layout() 
    plt.savefig('synthetic_2.png', dpi = 600) 
    plt.show()