如何将文本放在顶部的matplotlib 3D条形图中

时间:2015-03-29 10:38:24

标签: python graph matplotlib 3d

我需要为我的大学项目绘制一个3D条形图,条形图正在绘制,但我希望文本显示在每个条形图上,显示其z轴的值。

这是显示2D图表的代码段: source code link

#!/usr/bin/env python
# a bar plot with errorbars
import numpy as np
import matplotlib.pyplot as plt

N = 5
menMeans = (20, 35, 30, 35, 27)
menStd =   (2, 3, 4, 1, 2)

ind = np.arange(N)  # the x locations for the groups
width = 0.35       # the width of the bars

fig, ax = plt.subplots()
rects1 = ax.bar(ind, menMeans, width, color='r', yerr=menStd)

womenMeans = (25, 32, 34, 20, 25)
womenStd =   (3, 5, 2, 3, 3)
rects2 = ax.bar(ind+width, womenMeans, width, color='y', yerr=womenStd)

# add some text for labels, title and axes ticks
ax.set_ylabel('Scores')
ax.set_title('Scores by group and gender')
ax.set_xticks(ind+width)
ax.set_xticklabels( ('G1', 'G2', 'G3', 'G4', 'G5') )

ax.legend( (rects1[0], rects2[0]), ('Men', 'Women') )

def autolabel(rects):
    # attach some text labels
    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)

plt.show()

3D条形图的代码为source code for 3d bar-graph

在2D版本中,autolabel函数使用get_height(),get_x()和get_width()来动态计算高度,在3D版本中不支持:

  • 未找到功能的patch3d错误

是否有任何解决方法可以在2D图形中显示文本。

2 个答案:

答案 0 :(得分:2)

您可以直接从数据中提取条形坐标,并使用matplotlib.pyplot。text,如igavriil所指出的那样。对于您提供的示例,如下所示:

# Note that z and y-axis are swapped in the plot.
for x,y in zip(xs,ys):
    ax.text(x,z,y,'%2.2f'%y, horizontalalignment='left', verticalalignment='bottom')

答案 1 :(得分:0)

感谢igavriil该链接有所帮助,也感谢hr tur tur

我没有得到那个参数'%2.2f'%y它是如何工作的,可能有更清晰的方式来绘制这个但是我把它绘制成这样:

    def autolabel():
    """ attach some text labels"""
    for i in range(0, n):
        ax.text(int(x[i])+1, y[i]-2, z[i]+0.05, str(y[i])[0:4],     color='#660066', backgroundcolor= '#c187e1', weight= 'bold', rotation='vertical', fontsize=10, horizontalalignment='left', verticalalignment='bottom')      

链接:

DropBox link