python - 如何防止使用 matplotlib 对 python 条进行字母排序？

Question

我正在使用条形图绘制一些分类数据。即使我对数据框值进行排序，Matplotlib 仍会按字母顺序对 x 轴进行排序。 这是我的代码:

fig3, new_ax = plt.subplots(1,1, figsize=(25/3,5))
summary = tsa.source.sum().sort_values(ascending=False)
new_ax.bar(summary.index, summary.values, color=my_colors)
new_ax.legend()
bar_heights(new_ax) # custom function to get the values on top of bars
simpleaxis(new_ax) # custom function to define an axis to please my boss...
new_ax.set_ylabel('Effectifs')
new_ax.set_xlabel("Type d'arme")
new_ax.grid(False)

输出:

但这是摘要的样子，这就是我希望在图表上看到的顺序:

famas            2214.0
aut_typebruit     759.0
grena             200.0
flg                78.0
douze              72.0
sept               53.0
dtype: float64

这是我的数据样本的链接: https://files.fm/u/wumscb4q 用这一行导入它:

tsa.source = pd.read_csv('sample.csv', sep=';', index_col=0)

这里是我的功能:

def simpleaxis(ax):
    ax.spines['top'].set_visible(False)
    ax.spines['right'].set_visible(False)
    ax.get_xaxis().tick_bottom()
    ax.get_yaxis().tick_left()

def bar_heights(axes):
    for rect in axes.containers[0]:
        height = rect.get_height()
        rect.axes.text(rect.get_x() + rect.get_width()/2., height+3,
            '%d' % int(height),
            ha='center', va='bottom')

Answer 1

问题

这是 a bug在 matplotlib < 2.2.0 中，即使值是字符串，也始终对 X 轴进行排序。这就是下图中条形顺序颠倒的原因。

x = ['foo', 'bar']
y = [1, 2]
plt.bar(x, y, color=['b', 'g'])

修复

将 matplotlib 升级到 2.2.0 或更高版本。对于这些版本，相同的代码会产生预期的结果。

解决方法

如果您不能或不想升级 matplotlib，您可以使用数字而不是字符串，然后将刻度和标签设置为正确的值:

index = range(len(x))
plt.bar(x, y, color=['b', 'g'])  # use numbers in the X axis.
plt.xticks(index, x)  # set the X ticks and labels

Pandas 之道

您可以使用 Series.plot，这对您来说可能很方便，因为您的数据已经采用 Series 的形式，但请注意 pandas 以不同的方式绘制内容。使用关键字参数 rot 来控制标签的旋转。

s = pd.Series(y, index=x)
s.plot(kind='bar',rot=0, color=['b', 'g'])