The python code to draw s/b plot