我试图使用一个sankey图表来显示一些使用PySankey的用户细分变化，但类顺序与我想要的相反。有没有一种方法可以让我指定每个类的发布顺序？下面是我使用的代码（虚拟版本）：

test_df = pd.DataFrame({
    'curr_seg':np.repeat(['A','B','C','D'],4),
    'new_seg':['A','B','C','D']*4,
    'num_users':np.random.randint(low=10, high=20, size=16)
})

sankey(
    left=test_df["curr_seg"], right=test_df["new_seg"], 
    leftWeight= test_df["num_users"], rightWeight=test_df["num_users"], 
    aspect=20, fontsize=20
)

它产生了这个图表：

我想在左右轴上都先有A级，最后有D级。有谁知道我该怎么安装吗？非常感谢。