python - 在 Python 中创建一个数据透视表的图表

Question

我创建了一个数据透视表，我想创建一个条形图。这是我的数据透视表：

我不知道如何提取 1970 列的值并使用此信息制作条形图。

谢谢！！

标签： pythonpandasdataframedata-sciencedata-analysis

Answer 1

您可以使用 plt.bar 并切片数据框：

df = pd.DataFrame([[1970,'a',1],
                   [1970,'b',2], 
                   [1971,'a',2],
                   [1971,'b',3]],
                   columns=['year','location', 'value'])
df = pd.pivot_table(df, values='value', index='location', columns='year')
plt.bar(list(df.transpose().columns), height=df[1970])