def Af_to_stick(Af):
    """
    Given the final-state amplitudy Af, return a stick list
    Af: a dictionary of array([energy, amplitude])
    """
    return [ [Af[conf][0].real, float(abs(Af[conf][1]) ** 2)] for conf in Af]