Matplotlib: Adjusting the ticks to fit within the figure

问题

I have the following matplotlib code which all it does is plots 0-20 on the x-axis vs 0-100 on the y-axis

import matplotlib.pyplot as plt
fig = plt.figure()
ax = fig.add_subplot(111)
ax.plot(range(20))
ax.set_yticks(range(100))
labels = ax.set_yticklabels(range(100))

The problem I have is that not all the labels actually fit within the y-axis.

What I want to do is to divide the maximum label size into the total area of the figure to calculate how many labels I can put so that then I can group my data into sections 0-10, 10-20, 30-40 and plot those rather than display individual labels 1,2,3,4,5,6,7,8...

I am having trouble with finding the label size of the labels and the label size of the figure (correct me if I am wrong maybe it is the axis size in pixels. It is the first time I am using matplotlib)

How can I find out the number of pixels the figure/axes is taking up (width + height) and also the number of pixels (width + height) of each label?

回答1:

You could obtain the axis extents in your plot with ax.get_position(), see the documentation on axes (I would post a link, but my current reputation does not allow it).

The tick labelsize is defined in your matplotlibrc file relative to your font size. See some detail here. The default size is 12pt and can be changed with

plt.rcParams['ytick.labelsize'] = small

From the standard matplotibrc file:

Special text sizes can be defined relative to font.size, using the following values: xx-small, x-small, small, medium, large, x-large, xx-large, larger, or smaller

The easiest way to solve your problem is probably to just change the total figure size by using

fig = plt.figure(figsize=(10,30))

with the following result in your case (note that the example is a little bit extreme). If there are so many ticklabels that you have to increase the total figure size so much, you should consider decreasing the ticklabel size like shown above or decrease the number of ticklabels.

Note that this procedure involves manually adjusting the size of your total plot and/or your ticklabels until you obtain your desired output. If there is some better way to do it, I would be pleased to see it.

You can save your plot with plt.savefig(output.png) in your current working directory. If you use plt.show() and the image is larger than the popup-window, the labels will always be messed up, because the total figure is shrinked down to the window size but the label size stays constant.

I hope this answer is helpful for you.

回答2:

Simply add the following line:

plt.tight_layout()

回答3:

Another option is to push every other label slightly to the left. In addition to a smaller yticklabel size, this can look good:

import matplotlib as mpl
import matplotlib.pyplot as plt
import numpy as np

# Generate some random data
labels = 'abcdefghijklmnopqrstuvwxyz'
data = np.random.random((len(labels), 100))

# Plot it, setting the ticklabel size temporarily
# this is helpful for adjusting your plot until it's right
# without having an influence on other plots outside the context
with mpl.rc_context(rc={'ytick.labelsize': 6}):
    fig, ax = plt.subplots(1, 1, figsize=(6.5,5))
    im = ax.imshow(data)

    # Plot the labels on the yaxis
    ax.set_yticks(range(len(labels)))

    # Every second label gets a bit of whitespace to the right
    # thus effectively it is pushed to the left
    labels_formatted = [label if i%2==0 else label+' '*3 for i,label in enumerate(labels)]

    ax.set_yticklabels(labels_formatted)

来源：https://stackoverflow.com/questions/36252719/matplotlib-adjusting-the-ticks-to-fit-within-the-figure

标签

python

matplotlib

figure