python整数上限,Python Matplotlib限制为整数刻度位置

Quite often I want to make a bar chart of counts. If the counts are low I often get major and/or minor tick locations that are not integers. How can I prevent this? It makes no sense to have a tick at 1.5 when the data are counts.

This is my first attempt:

import pylab

pylab.figure()

ax = pylab.subplot(2, 2, 1)

pylab.bar(range(1,4), range(1,4), align='center')

major_tick_locs = ax.yaxis.get_majorticklocs()

if len(major_tick_locs) < 2 or major_tick_locs[1] - major_tick_locs[0] < 1:

ax.yaxis.set_major_locator(pylab.MultipleLocator(1))

minor_tick_locs = ax.yaxis.get_minorticklocs()

if len(minor_tick_locs) < 2 or minor_tick_locs[1] - minor_tick_locs[0] < 1:

ax.yaxis.set_minor_locator(pylab.MultipleLocator(1))

which works OK when the counts are small but when they are large, I get many many minor ticks:

import pylab

ax = pylab.subplot(2, 2, 2)

pylab.bar(range(1,4), range(100,400,100), align='center')

major_tick_locs = ax.yaxis.get_majorticklocs()

if len(major_tick_locs) < 2 or major_tick_locs[1] - major_tick_locs[0] < 1:

ax.yaxis.set_major_locator(pylab.MultipleLocator(1))

minor_tick_locs = ax.yaxis.get_minorticklocs()

if len(minor_tick_locs) < 2 or minor_tick_locs[1] - minor_tick_locs[0] < 1:

ax.yaxis.set_minor_locator(pylab.MultipleLocator(1))

How can I get the desired behaviour from the first example with small counts whilst avoiding what happens in the second?

解决方案

You can use the MaxNLocator method, like so:

from pylab import MaxNLocator

ya = axes.get_yaxis()

ya.set_major_locator(MaxNLocator(integer=True))