Calculate probability of neutron porosity from drilling torque

idm2.py

import statviz

# Choose lithology
lith = 'marl' # sandstone, claystone, marl

# Two variables to plot against
x, y = 'TORQUE', 'BPHI'

# Bounds of each variable to evaluate probabilities
xbound, ybound = (20e3, 22.5e3), (20, 30)

df = merge_fac_df[merge_fac_df.LITH==lith]

# Plot KDE
statviz.kde2d(df, x, y, evaluate_proba=True, xbound=xbound, ybound=ybound)

# Calculate probabilities from KDE
Px, Py, Pxny, Pxy_cond = statviz.compute_proba_2d(df, x, y, xbound, ybound)
print(Px, Py, Pxny, Pxy_cond)