如何在python中显示条形图

时间:2019-07-09 10:15:01

标签: python python-3.x pandas matplotlib bar-chart

我正在尝试用Python创建带有多个条形图的条形图。条形图应在每个条形图的顶部显示值。 我有一个如下数据集:

Speciality                  Very interested Somewhat_interested Notinterested
Big Data (Spark/Hadoop)         1332           729                      127
Data Analysis / Statistics      1688           444                      60     
Data Journalism                 429            1081                     610

我尝试了以下代码:

import numpy as np
import pandas as pd
import matplotlib as mpl
import matplotlib.pyplot as plt

pd_dataframe = pd.read_csv('Test-Barchart.csv')
no_of_xaxis = pd_dataframe.Speciality.nunique()
ind = np.arange(no_of_xaxis)
xcord = pd_dataframe['Speciality'].tolist()

veryinterestedlist = pd_dataframe['Very interested'].tolist()
somewhatlist = pd_dataframe['Somewhat interested'].tolist()
notinterestedlist = pd_dataframe['Not interested'].tolist()

fig=plt.figure()
ax = fig.add_subplot(111)
width=0.8

rects1 = ax.bar(ind, veryinterestedlist, width, color='r')
rects2 = ax.bar(ind, somewhatlist, width, color='g')
rects3 = ax.bar(ind+width*2, notinterestedlist, width, color='b')

ax.legend( (rects1[0], rects2[0], rects3[0]), ('Very Interested', 
 'Somewhat Interested', 'Not Interested') )

def autolabel(rects):
   for rect in rects:
    h = rect.get_height()
    ax.text(rect.get_x()+rect.get_width()/2., 1.05*h, '%d'%int(h),
            ha='center', va='bottom')

autolabel(rects1)
autolabel(rects2)
autolabel(rects3)

ax.set_xticks(ind+width)
ax.set_xticklabels( xcord )
plt.show()

问题是plt.show()没有显示任何内容! 我在代码中没有任何错误。 您能帮我解决这个问题吗? 另外我如何将条形颜色更改为十六进制代码颜色而不是r,g或b?例如#5bc0de

1 个答案:

答案 0 :(得分:-1)

对代码的小的更改:

import pandas as pd
import matplotlib.pyplot as plt
import numpy as np

pd_dataframe = pd.read_csv('Test-Barchart.csv')
no_of_xaxis = pd_dataframe.Speciality.nunique()
ind = np.arange(no_of_xaxis)
width = 0.1
xcord = pd_dataframe['Speciality'].tolist()

veryinterestedlist = pd_dataframe['Very interested'].tolist()
somewhatlist = pd_dataframe['Somewhat interested'].tolist()
notinterestedlist = pd_dataframe['Not interested'].tolist()

fig, ax = plt.subplots()

rects1 = ax.bar(ind, veryinterestedlist, width, color='g')
rects2 = ax.bar(ind + width, somewhatlist, width, color='c')
rects3 = ax.bar(ind+2*width, notinterestedlist, width, color='r')

# add some text for labels, title and axes ticks
ax.set_ylabel('y label')
ax.set_title('Title')
ax.set_xticks(ind + width)
ax.set_xticklabels(xcord)

ax.legend( (rects1[0], rects2[0], rects3[0]), ('Very Interested', 
 'Somewhat Interested', 'Not Interested') )

def autolabel(rects):
    """
    Attach a text label above each bar displaying its height
    """
    for rect in rects:
        height = rect.get_height()
        ax.text(rect.get_x() + rect.get_width()/2., 1.05*height,
                '%d' % int(height),
                ha='center', va='bottom')

autolabel(rects1)
autolabel(rects2)
autolabel(rects3)

plt.show()

您会得到:

enter image description here

参考:Grouped bar chart with labels