python实现利用透视表创建条形图并添加数字
时间: 2023-07-30 07:07:12 浏览: 195
好的,我可以回答您的问题。首先,我们需要使用pandas库中的pivot_table方法创建透视表,然后使用matplotlib库的bar方法创建条形图,并使用text方法添加数字。下面是一个示例代码:
``` python
import pandas as pd
import matplotlib.pyplot as plt
# 创建数据框
df = pd.DataFrame({'Gender': ['M', 'M', 'F', 'F', 'M', 'F', 'M', 'F'],
'Age': [20, 30, 25, 35, 28, 40, 32, 27],
'Salary': [30000, 50000, 45000, 55000, 40000, 60000, 55000, 50000]})
# 创建透视表
pivot_table = df.pivot_table(index='Gender', values='Salary', aggfunc='mean')
# 创建条形图并添加数字
plt.bar(pivot_table.index, pivot_table['Salary'], color='blue')
for i, v in enumerate(pivot_table['Salary']):
plt.text(i, v+1000, str(v), horizontalalignment='center')
# 设置图形标题和标签
plt.title('Average Salary by Gender')
plt.xlabel('Gender')
plt.ylabel('Salary')
# 显示图形
plt.show()
```
在这个示例中,我们创建了一个包含Gender、Age和Salary列的数据框。然后,我们使用pivot_table方法创建了一个根据Gender列分组并计算Salary均值的透视表。接下来,我们使用bar方法创建了一个条形图,并使用text方法添加了每个条形上的数字。最后,我们设置了图形的标题和标签,并使用show方法显示了图形。
阅读全文