matplotlib的“barh”中的高度变量单位是多少?

时间:2016-08-05 16:00:29

标签: python matplotlib

在matplotlib函数barh的定义中:

matplotlib.pyplot.barh(bottom, width, height=0.8, left=None, hold=None, **kwargs)

默认的“高度”是0.8,但是当我绘制一些具有不同图形高度的图形时,例如(30,40,..)和dpi = 100。我看到酒吧的高度发生了变化。这不是固定的。所以我想知道什么是高度的单位,以及如何使它固定(不依赖于图的高度)。

1 个答案:

答案 0 :(得分:4)

我将其分为两部分:

  

我想知道什么是高度的单位

(显然人们一直在想这个since 2009 ......所以我猜你是一个很好的公司!)

这个问题是更容易的部分 - 它是分配给图中条形图的高度的百分比。例如,默认height=0.8表示条形的高度为0.8 * (plot_height / n_bars)。您可以通过设置height=1.0(或者甚至值> 1,条形图将重叠)来查看此内容。

如果你真的想确定,请source of axes.barh。这只是致电axes.bar - 请查看these lines

nbars = len(bottom)
if len(left) == 1:
    left *= nbars
if len(height) == 1:
    height *= nbars

later on ......

args = zip(left, bottom, width, height, color, edgecolor, linewidth)
for l, b, w, h, c, e, lw in args:
    if h < 0:
        b += h
        h = abs(h)
    if w < 0:
        l += w
        w = abs(w)
    r = mpatches.Rectangle(
        xy=(l, b), width=w, height=h,
        facecolor=c,
        edgecolor=e,
        linewidth=lw,
        label='_nolegend_',
        margins=margins
        )
    r.update(kwargs)
    r.get_path()._interpolation_steps = 100
    #print r.get_label(), label, 'label' in kwargs
    self.add_patch(r)
    patches.append(r)

因此,您可以看到高度按nbars缩放,当您绘制矩形时,它们会按此高度间隔开。

  

如何修复

这个更难,你必须手动设置它。图表上的条形图最终是matplotlib.patches.Rectangle个对象,它们具有宽度和高度......这也是一个百分比。我认为最好的解决方案是手动计算适当的百分比。

这是一个简短的例子,基于barh demo

import matplotlib.pyplot as plt
plt.rcdefaults()
import numpy as np
import matplotlib.pyplot as plt

# Example data
people = ('Tom', 'Dick', 'Harry', 'Slim', 'Jim')
y_pos = np.arange(len(people))
performance = 3 + 10 * np.random.rand(len(people))
error = np.random.rand(len(people))

plt.figure(figsize=(5,5), dpi=80)
myplot = plt.barh(y_pos, performance, height=0.8, xerr=error, align='center', alpha=0.4)
plt.yticks(y_pos, people)
plt.xlabel('Performance')
plt.title('How fast do you want to go today?')

for obj in myplot:
    # Let's say we want to set height of bars to always 5px..
    desired_h = 5
    current_h = obj.get_height()
    current_y = obj.get_y()
    pixel_h = obj.get_verts()[2][1] - obj.get_verts()[0][1]
    print("current position = ", current_y)
    print("current pixel height = ", pixel_h)

    # (A) Use ratio of pixels to height-units to calculate desired height
    h = desired_h / (pixel_h/current_h)
    obj.set_height(h)
    pixel_h = obj.get_verts()[2][1] - obj.get_verts()[0][1]
    print("now pixel height = ", pixel_h)

    # (B) Move the rectangle so it still aligns with labels and error bars
    y_diff = current_h - h # height is same units as y
    new_y = current_y + y_diff/2
    obj.set_y(new_y)
    print("now position = ", obj.get_y())
plt.show()

A部分计算pixel_h/current_h以获得像素和高度单位之间的转换。然后,我们可以将desired_h(像素)除以该比率,以获得高度单位的desired_h。这会将条形宽度设置为5像素,但条形图的底部保持在同一位置,因此它不再与标签和误差线对齐。

B部分计算新的y位置。由于yheight属于相同的单位,我们只需加一半高度差(y_diff)即可获得新位置。这使得栏杆以任何原始的y位置为中心。

请注意,这仅设置初始大小。例如,如果您调整绘图的大小,条形图仍将缩放 - 您必须覆盖该事件以适当调整条形大小。