if pp == 0:
fig2 = plt.figure(figsize=(7, 7))
# Second plot for the predicted apparent resistivity data
ax2 = fig2.add_subplot(2, 1, pp + 1, aspect='equal')
ax2.pcolor(mesh2d.vectorNx,
mesh2d.vectorNy,
np.log10(m2D),
edgecolor="none",
alpha=0.5,
cmap='gray')
# Add the speudo section
dat = DC.plot_pseudoSection(survey2D,
ax2,
stype=stype,
clim=[-2.5, -0.5],
dtype=dtype)
if pp == 0:
circle = plt.Circle((12000, 135), 100, color='k', fill=False, lw=3)
ax2.add_artist(circle)
ax2.set_xlim([xmin, xmax])
ax2.set_ylim([zmin, zmax])
plt.show()
if pp == 0:
ax2.get_xaxis().set_ticks([])
#%% Load model and predicted data
minv = DC.readUBC_DC2DModel(inv_dir + dsep + 'dcinv2d.con')
minv = np.reshape(minv, (mesh2d.nCy, mesh2d.nCx))
dpre = DC.readUBC_DC2Dpre(inv_dir + dsep + 'dcinv2d.pre')
DCpre = dpre['DCsurvey']
DCtemp = survey
DCtemp.dobs = DCpre.dobs
if kk == 0:
ax5 = plt.subplot(3, 2, 3)
DC.plot_pseudoSection(DCtemp,
ax5,
surveyType='pole-dipole',
clim=(ph.get_clim()[0], ph.get_clim()[1]),
colorbar=False)
ax5.set_title('Predicted', fontsize=10)
plt.xlim([xmin, xmax])
plt.ylim([zmin, zmax])
plt.gca().set_aspect('equal', adjustable='box')
ax5.set_xticklabels([])
ax5.set_yticks(map(int, z))
ax5.set_yticklabels(map(str, map(int, z)), rotation='vertical')
ax5.set_ylabel('Depth (m)', fontsize=8)
else:
ax6 = plt.subplot(3, 2, 4)
DC.plot_pseudoSection(DCtemp,
