matplotlib 如何绘制和注解分组条形图

jdzmm42g 于 2023-05-23 发布在其他

关注(0)|答案(1)|浏览(125)

我遇到了一个关于Python中matplotlib的棘手问题。我想创建一个有几个代码的分组条形图，但图表出错了。你能给我提些建议吗？代码如下。

import numpy as np
import pandas as pd
file="https://s3-api.us-geo.objectstorage.softlayer.net/cf-courses-data/CognitiveClass/DV0101EN/labs/coursera/Topic_Survey_Assignment.csv"
df=pd.read_csv(file,index_col=0)

df.sort_values(by=['Very interested'], axis=0,ascending=False,inplace=True)

df['Very interested']=df['Very interested']/2233
df['Somewhat interested']=df['Somewhat interested']/2233
df['Not interested']=df['Not interested']/2233
df

df_chart=df.round(2)
df_chart

labels=['Data Analysis/Statistics','Machine Learning','Data Visualization',
       'Big Data (Spark/Hadoop)','Deep Learning','Data Journalism']
very_interested=df_chart['Very interested']
somewhat_interested=df_chart['Somewhat interested']
not_interested=df_chart['Not interested']

x=np.arange(len(labels))
w=0.8

fig,ax=plt.subplots(figsize=(20,8))
rects1=ax.bar(x-w,very_interested,w,label='Very interested',color='#5cb85c')
rects2=ax.bar(x,somewhat_interested,w,label='Somewhat interested',color='#5bc0de')
rects3=ax.bar(x+w,not_interested,w,label='Not interested',color='#d9534f')

ax.set_ylabel('Percentage',fontsize=14)
ax.set_title("The percentage of the respondents' interest in the different data science Area",
            fontsize=16)
ax.set_xticks(x)
ax.set_xticklabels(labels)
ax.legend(fontsize=14)

def autolabel(rects):
    """Attach a text label above each bar in *rects*, displaying its height."""
    for rect in rects:
        height = rect.get_height()
        ax.annotate('{}'.format(height),
                    xy=(rect.get_x() + rect.get_width() / 3, height),
                    xytext=(0, 3),  # 3 points vertical offset
                    textcoords="offset points",
                    ha='center', va='bottom')

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

fig.tight_layout()

plt.show()

这个代码模块的输出真是一团糟。但我所期望的应该看起来像图片中的条形图。你能告诉我我的代码中哪一点不正确吗？

matplotlib

来源：https://stackoverflow.com/questions/63493530/how-to-plot-and-annotate-a-grouped-bar-chart

1条答案

按热度按时间

xiozqbni1#

导入和DataFrame

import pandas as pd
import matplotlib.pyplot as plt

# given the following code to create the dataframe
file="https://s3-api.us-geo.objectstorage.softlayer.net/cf-courses-data/CognitiveClass/DV0101EN/labs/coursera/Topic_Survey_Assignment.csv"
df=pd.read_csv(file, index_col=0)

df.sort_values(by=['Very interested'], axis=0, ascending=False, inplace=True)

# all columns are being divided by 2233 so those lines can be replace with the following single line
df = df.div(2233)

# display(df)
                            Very interested  Somewhat interested  Not interested
Data Analysis / Statistics         0.755934             0.198836        0.026870
Machine Learning                   0.729512             0.213614        0.033139
Data Visualization                 0.600090             0.328706        0.045678
Big Data (Spark / Hadoop)          0.596507             0.326467        0.056874
Deep Learning                      0.565607             0.344828        0.060905
Data Journalism                    0.192118             0.484102        0.273175

使用since `matplotlib v3.4.2`

使用matplotlib.pyplot.bar_label和pandas.DataFrame.plot
一些格式化可以使用fmt参数完成，但更复杂的格式化应该使用labels参数完成，如How to add multiple annotations to a barplot所示。
有关使用.bar_label的其他详细信息和示例，请参见How to add value labels on a bar chart

# your colors
colors = ['#5cb85c', '#5bc0de', '#d9534f']

# plot with annotations is probably easier
ax = df.plot(kind='bar', color=colors, figsize=(20, 8), rot=0, ylabel='Percentage', title="The percentage of the respondents' interest in the different data science Area")

for c in ax.containers:
    ax.bar_label(c, fmt='%.2f', label_type='edge')

在`matplotlib v3.4.2`之前使用

w = 0.8 / 3将解决这个问题，给定当前代码。
但是，使用pandas.DataFrame.plot可以更轻松地生成绘图

# your colors
colors = ['#5cb85c', '#5bc0de', '#d9534f']

# plot with annotations is probably easier
ax = df.plot.bar(color=colors, figsize=(20, 8), ylabel='Percentage', title="The percentage of the respondents' interest in the different data science Area")
ax.set_xticklabels(ax.get_xticklabels(), rotation=0)

for p in ax.patches:
    ax.annotate(f'{p.get_height():0.2f}', (p.get_x() + p.get_width() / 2., p.get_height()), ha = 'center', va = 'center', xytext = (0, 10), textcoords = 'offset points')

赞(0）回复(0）举报 2023-05-23

我来回答

matplotlib 如何绘制和注解分组条形图

1条答案

导入和DataFrame

使用since `matplotlib v3.4.2`

在`matplotlib v3.4.2`之前使用

相关问题

热门标签

最新问答

matplotlib 如何绘制和注解分组条形图

1条答案

导入和DataFrame

使用since matplotlib v3.4.2

在matplotlib v3.4.2之前使用

相关问题

热门标签

最新问答

使用since `matplotlib v3.4.2`

在`matplotlib v3.4.2`之前使用