使用python和matplotlib获取打印在条形图顶部的值?

时间:2019-05-09 07:59:48

标签: python python-3.x matplotlib

使用python和matplotlib获取打印在条形图顶部的值吗?

我正在使用以下代码:

import pandas as pd
import matplotlib.pyplot as plt
hsif = pd.read_excel('C:\\Users\\ans\\Desktop\\test.xlsx')
class_sum = hsif.groupby('Class')['Value'].sum()
class_sum_sort = class_sum.sort_values(ascending=False)
ax = class_sum_sort.plot.barh(x='Class',y=class_sum_sort).invert_yaxis()
plt.title('My Dynamic Graphs')
plt.xlabel('Value',fontsize=20)
plt.ylabel('Class',fontsize=20)
plt.show()

Excel文件包含的数据为

Class   Value
Six 3763
Six 2570
Six 1123
Six 901
Seven   12210
Seven   10384
Seven   3188
Seven   2592
Seven   1723
Seven   1011

因此,我尝试打印该图,其顶部的值会根据图表自动更改。我们还可以添加特定于六种颜色的橙色和七种红色的颜色。 谁能帮我这个忙。

1 个答案:

答案 0 :(得分:-1)

您可以遍历栏并设置文本和颜色。

import pandas as pd
import matplotlib.pyplot as plt

hsif = pd.read_excel('C:\\Users\\ans\\Desktop\\test.xlsx')
class_sum = hsif.groupby('Class')['Value'].sum()
class_sum_sort = class_sum.sort_values(ascending=False)
ax = class_sum_sort.plot(kind="barh", fontsize=13)
ax.set_title('My Dynamic Graphs')
ax.set_xlabel('Value', fontsize=20)
ax.set_ylabel('Class', fontsize=20)    

for i in ax.patches:
    ax.text(i.get_width(), i.get_y()+.3, str(i.get_width()), fontsize=13)

ax.patches[0].set_color("red")
ax.patches[1].set_color("orange")

ax.invert_yaxis()
plt.tight_layout()
plt.show()