python - Pandas：图中的分组和 xticks

Question

我正在使用具有以下结构的同一日期：

我想按列对数据进行分组B，并获取绘图和比较的平均值。

sub_data = data_composite.groupby(['B']).aggregate(np.mean)
ax = sub_data.plot()

获得：

但是，我想得到图中对应的 xticks。它将是 KP40、KP08 等......像这样的东西：

有没有办法做到这一点？

非常感谢。亲切的问候，

标签： pythonpandasmatplotlibplot

Answer 1

它应该适合你

import numpy as np
import matplotlib.pyplot as plt

tmp_labels = data_composite.drop_duplicates(subset='B', keep = 'first')
xlabels = tmp_labels['B'].values
plt.xticks(np.arange(sub_data.shape[0]),list(xlabels), rotation=90)