A function to visualize probabilities against state number (like what's done in the pulser example notebook).