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,