if ax is None:
    ax = plt.gca()
    fig = ax.figure
    ax.set_title('Root Locus')
else:
    fig = plt.gcf()

or fig = ax.figure