tukey_scatter.py

def tukey_scatter(x, y, iqrs=1.5, ax=None, **kwargs):
    ax = ax or plt.gca()
    x_25, x_75 = np.percentile(x, [25, 75])
    x_iqr = x_75 - x_25
    y_25, y_75 = np.percentile(x, [25, 75])
    y_iqr = y_75 - y_25
    inliers = np.flatnonzero(
        (x_25 - iqrs * x_iqr < x)
        & (x < x_75 + iqrs * x_iqr)
        & (y_25 - iqrs * y_iqr < y)
        & (y < y_75 + iqrs * y_iqr)
    )
    for k, v in kwargs.items():
        if isinstance(v, np.ndarray) and v.shape == x.shape:
            kwargs[k] = v[inliers]
    return ax.scatter(x[inliers], y[inliers], **kwargs)