Matplotlib: non-alignment of the dots on a plot

Question

I am using matplotlib to do a Component-Component plus Residual (CCPR) Plots (= partial residual plot)

This script :

fig, ax = plt.subplots(figsize=(5, 5))
fig = sm.graphics.plot_ccpr(lm_full, 'diag[T.sz]', ax=ax)
plt.close

Gives :

How can I modify my script to get something like

I don't want my dots to be aligned. In both cases, the variables of the x axis are dummy variable (ill vs healthy controls).

This may seem stupid, but I don't even know how to express what I want : it's much more easier with the images.

Answer 1

It sounds like you want to add some jitter to the x values, like this:

import numpy as np

# get x and y coordinates from the axes
coords = ax.collections[0].get_offsets()

# add small random number to each x coordinate
coords[:,0] = coords[:,0] + np.random.rand(coords.shape[0]) * 0.01

# move the points to the new coordinates
ax.collections[0].set_offsets(coords)