python - 如何创建带类别的 Matplotlib 条形图？

Question

我正在尝试创建一个在 X 轴上带有类别的 matplotlib 条形图，但我无法获得正确的类别。这是我正在尝试做的事情的一个最小示例。

data = [[46, 11000], [97, 15000], [27, 24000], [36, 9000], [9, 17000]]
df = pd.DataFrame(data, columns=['car_id', 'price'])

fig1, ax1 = plt.subplots(figsize=(10,5))
ax1.set_title('Car prices')
ax1.bar(df['car_id'], df['price'])
plt.xticks(np.arange(len(df)), list(df['car_id']))
plt.legend()
plt.show()

我需要 X 轴上的五个类别 (car_id)。我究竟做错了什么？ :-/

Answer 1

你可以把car_id变成category:

df['car_id'] = df['car_id'].astype('category')

df.plot.bar(x='car_id')

输出:

您也可以只绘制 price 列并重新标记:

ax = df.plot.bar(y='price')
ax.set_xticklabels(df['car_id'])