使用matplotlib自动设置条形图的y轴限制

时间:2012-06-26 21:36:32

标签: python matplotlib

以下是绘制大值的示例。

import matplotlib.pyplot as plt
x = [1,2,3,4,5]
y = [1000, 1002, 1001, 1003, 1005]
plt.bar(x,y) 
plt.show()

y轴从0开始,所以条形看起来都相等。我知道你可以使用plt.ylim来手动设置限制,但matplotlib是否有办法自动(并巧妙地)将限制设置为合理的值(如998-1008),也可能显示轴断点?

1 个答案:

答案 0 :(得分:20)

一些简单的代数将有助于修复限制:

import matplotlib.pyplot as plt
import math
x = [1,2,3,4,5]
y = [1000, 1002, 1001, 1003, 1005]
low = min(y)
high = max(y)
plt.ylim([math.ceil(low-0.5*(high-low)), math.ceil(high+0.5*(high-low))])
plt.bar(x,y) 
plt.show()

通过这种方式,您可以找到y值之间的差异,并使用它们来设置沿y轴的比例。我使用math.ceil(而不是math.floor)来获取您指定的值并确保整数。

就轴心而言,我建议查看this example