Little Python script to create seismic polarity cartoons.

seismic_polarity_cartoons.py

import numpy as np
import matplotlib.pyplot as plt
import bruges as b

def disegna_schemino(z, ai, wavelet,titolo):
    f, ax = plt.subplots(nrows=1,ncols=2, figsize=(2,2),facecolor='w')
    ax[0].plot(ai,z, '-k', lw=4)
    ax[1].plot(wavelet, z, '-k', lw=2)
    if colori:
        ax[1].fill_betweenx(z,wavelet,0, where=wavelet>0, facecolor=[0.6,0.6,1.0], linewidth=0)
        ax[1].fill_betweenx(z,wavelet,0, where=wavelet<0, facecolor=[1.0,0.7,0.7], linewidth=0)
    else:
        ax[1].fill_betweenx(z,wavelet,0, where=wavelet>0, facecolor='black', linewidth=0)
    ax[0].set_title('AI', fontsize=8)
    ax[1].set_title(titolo, fontsize=8)
    ax[0].set_xlim(-1.4,1.4)
    ax[1].set_xlim(-1.2,1.2)
    for aa in ax:
        aa.set_ylim(50,200)
        aa.invert_yaxis()
        aa.set_yticklabels([])
        aa.set_xticklabels([])
        aa.xaxis.set_ticks_position('none')
        aa.yaxis.set_ticks_position('none')

wavelet=b.filters.ricker(0.25, 0.001, 25)
ai=np.zeros(wavelet.size)
z=np.arange(wavelet.size)
ai[int(wavelet.size/2):]=1

disegna_schemino(z, ai, wavelet, titolo='SEG-normal')
plt.savefig('SEG-normal.png', dpi=300, bbox_inches='tight')

wavelet=b.filters.rotate_phase(wavelet, np.radians(180))
disegna_schemino(z, ai, wavelet, titolo='SEG-reverse')
plt.savefig('SEG-reverse.png', dpi=300, bbox_inches='tight')