首页 > 解决方案 > pandas 使用分类字符串值对 x 轴进行排序

问题描述

我正在尝试对熊猫图 x 轴中的分类变量进行排序。我尝试从熊猫数据框本身进行排序,但我只能按字母顺序排序。我没有在情节中看到以这种方式排序的选项。我正在尝试按以下顺序获取 x 轴: [Mon, Tue, Wed, Thru, Fri, Sat, Sun] 。我怎么做?

df = data_df.groupby(['name1','weekday']).sum().reset_index()
s = (df.pivot_table(
        index='weekday', columns='name1', values='count', aggfunc='sum'))

s.plot(kind='bar', stacked=True,figsize=(20,10),rot=0)
plt.show()

在此处输入图像描述

寻找这个输出

标签: python-3.xpandasplotly

解决方案


据我所知,你的dfs本质上是一样的。此外,您可以reindex然后绘制:

weekdays = ['Monday','Tuesday','Wednesday','Thursday',
            'Friday','Saturday','Sunday']
# sample data
np.random.seed(1)
data_df = pd.DataFrame({
    'name1':np.random.randint(0,3, 100),
    'weekday':np.random.choice(weekdays, 100),
    'count':np.random.randint(3,100)
})

# groupby and take sum
s = (data_df.groupby(['name1','weekday'])
       .sum().unstack('name1')
       .reindex(weekdays)                 # here is the key to get the correct order
)

# plot
s.plot.bar(stacked=True)

输出:

在此处输入图像描述


推荐阅读